hex.genmodel.algos.upliftdrf.UpliftDrfMojoReader Maven / Gradle / Ivy
package hex.genmodel.algos.upliftdrf;
import hex.genmodel.algos.tree.SharedTreeMojoReader;
import java.io.IOException;
/**
*/
public class UpliftDrfMojoReader extends SharedTreeMojoReader {
@Override
public String getModelName() {
return "Distributed Uplift Random Forest";
}
@Override
protected void readModelData() throws IOException {
super.readModelData();
_model._treatmentColumn = readkv("treatment_column");
_model._thresholds = readkv("thresholds");
}
@Override
protected UpliftDrfMojoModel makeModel(String[] columns, String[][] domains, String responseColumn) {
return null;
}
@Override
protected UpliftDrfMojoModel makeModel(String[] columns, String[][] domains, String responseColumn, String treatmentColumn) {
return new UpliftDrfMojoModel(columns, domains, responseColumn, treatmentColumn);
}
@Override public String mojoVersion() {
return "1.40";
}
}