All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.alibaba.alink.pipeline.regression.DecisionTreeRegressor Maven / Gradle / Ivy

package com.alibaba.alink.pipeline.regression;

import com.alibaba.alink.operator.batch.BatchOperator;
import com.alibaba.alink.operator.batch.regression.DecisionTreeRegTrainBatchOp;
import org.apache.flink.ml.api.misc.param.Params;
import com.alibaba.alink.params.regression.DecisionTreeRegPredictParams;
import com.alibaba.alink.params.regression.DecisionTreeRegTrainParams;
import com.alibaba.alink.pipeline.Trainer;

public class DecisionTreeRegressor extends Trainer  implements
	DecisionTreeRegTrainParams , DecisionTreeRegPredictParams  {

	public DecisionTreeRegressor() {
		super();
	}

	public DecisionTreeRegressor(Params params) {
		super(params);
	}

	@Override
	protected BatchOperator train(BatchOperator in) {
		return new DecisionTreeRegTrainBatchOp(this.getParams()).linkFrom(in);
	}

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy