com.alibaba.alink.params.regression.AftRegPredictParams Maven / Gradle / Ivy

package com.alibaba.alink.params.regression;

import com.alibaba.alink.params.shared.colname.*;
import org.apache.flink.ml.api.misc.param.ParamInfo;
import org.apache.flink.ml.api.misc.param.ParamInfoFactory;

/**
 * Params for AftRegressionPredictor.
 */
public interface AftRegPredictParams
        extends HasReservedCols,
        HasPredictionCol,
        HasVectorColDefaultAsNull,
        HasPredictionDetailCol {

    ParamInfo QUANTILE_PROBABILITIES = ParamInfoFactory
            .createParamInfo("quantileProbabilities", double[].class)
            .setDescription("Array of quantile probabilities.")
            .setHasDefaultValue(new double[]{0.01, 0.05, 0.1, 0.25, 0.5, 0.75, 0.9, 0.95, 0.99})
            .build();

    default double[] getQuantileProbabilities() {
        return get(QUANTILE_PROBABILITIES);
    }

    default T setQuantileProbabilities(double[] value) {
        return set(QUANTILE_PROBABILITIES, value);
    }

}