• Home
• com.alibaba.alink
• alink_core
• 1.0.1
• source code
• DecisionTreeTrainBatchOp.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.alibaba.alink.operator.batch.classification.DecisionTreeTrainBatchOp Maven / Gradle / Ivy

package com.alibaba.alink.operator.batch.classification;

import org.apache.flink.ml.api.misc.param.Params;

import com.alibaba.alink.operator.common.tree.BaseRandomForestTrainBatchOp;
import com.alibaba.alink.params.classification.DecisionTreeTrainParams;
import com.alibaba.alink.params.classification.RandomForestTrainParams;
import com.alibaba.alink.params.shared.tree.HasFeatureSubsamplingRatio;
import com.alibaba.alink.params.shared.tree.HasSubsamplingRatio;

/**
 * Fit a single decision tree.
 *
 * @see BaseRandomForestTrainBatchOp
 */
public final class DecisionTreeTrainBatchOp extends BaseRandomForestTrainBatchOp
	implements DecisionTreeTrainParams {

	public DecisionTreeTrainBatchOp() {
		this(null);
	}

	public DecisionTreeTrainBatchOp(Params params) {
		super(params);
		getParams().set(RandomForestTrainParams.NUM_TREES, 1);
		getParams().set(HasFeatureSubsamplingRatio.FEATURE_SUBSAMPLING_RATIO, 1.0);
		getParams().set(HasSubsamplingRatio.SUBSAMPLING_RATIO, 1.0);
	}
}