com.alibaba.alink.params.classification.GbdtTrainParams Maven / Gradle / Ivy

package com.alibaba.alink.params.classification;

import org.apache.flink.ml.api.misc.param.ParamInfo;
import org.apache.flink.ml.api.misc.param.ParamInfoFactory;

import com.alibaba.alink.params.shared.colname.HasGroupColDefaultAsNull;
import com.alibaba.alink.params.shared.tree.HasFeatureSubsamplingRatio;
import com.alibaba.alink.params.shared.tree.HasFeatureSubsamplingRatioDefaultAs1;
import com.alibaba.alink.params.shared.tree.HasMaxBins;
import com.alibaba.alink.params.shared.tree.HasMaxDepthDefaultAs6;
import com.alibaba.alink.params.shared.tree.HasMinSamplesPerLeafDefaultAs100;
import com.alibaba.alink.params.shared.tree.HasNumTreesDefaultAs100;
import com.alibaba.alink.params.shared.tree.HasSubsamplingRatio;
import com.alibaba.alink.params.shared.tree.HasSubsamplingRatioDefaultAs1;
import com.alibaba.alink.params.shared.tree.TreeTrainParams;

public interface GbdtTrainParams extends
	ClassifierTrainParams ,
	TreeTrainParams ,
	HasNumTreesDefaultAs100,
	HasMinSamplesPerLeafDefaultAs100,
	HasMaxDepthDefaultAs6,
	HasSubsamplingRatioDefaultAs1,
	HasFeatureSubsamplingRatioDefaultAs1,
	HasGroupColDefaultAsNull ,
	HasMaxBins  {

	ParamInfo  LEARNING_RATE = ParamInfoFactory
		.createParamInfo("learningRate", Double.class)
		.setDescription("learning rate for gbdt training(default 0.3)")
		.setHasDefaultValue(0.3)
		.build();
	ParamInfo  MIN_SUM_HESSIAN_PER_LEAF = ParamInfoFactory
		.createParamInfo("minSumHessianPerLeaf", Double.class)
		.setDescription("minimum sum hessian for each leaf")
		.setHasDefaultValue(0.0)
		.build();

	default Double getLearningRate() {
		return get(LEARNING_RATE);
	}

	default T setLearningRate(Double value) {
		return set(LEARNING_RATE, value);
	}

	default Double getMinSumHessianPerLeaf() {
		return get(MIN_SUM_HESSIAN_PER_LEAF);
	}

	default T setMinSumHessianPerLeaf(Double value) {
		return set(MIN_SUM_HESSIAN_PER_LEAF, value);
	}
}