hex.genmodel.algos.gbm.GbmMojoReader Maven / Gradle / Ivy
package hex.genmodel.algos.gbm;
import hex.genmodel.algos.tree.SharedTreeMojoReader;
import hex.genmodel.utils.DistributionFamily;
import java.io.IOException;
/**
*/
public class GbmMojoReader extends SharedTreeMojoReader {
@Override
public String getModelName() {
return "Gradient Boosting Machine";
}
@Override
protected void readModelData() throws IOException {
super.readModelData();
_model._family = DistributionFamily.valueOf((String)readkv("distribution"));
_model._init_f = readkv("init_f");
_model._link_function = readLinkFunction((String) readkv("link_function"), _model._family);
}
@Override
protected GbmMojoModel makeModel(String[] columns, String[][] domains, String responseColumn) {
return new GbmMojoModel(columns, domains, responseColumn);
}
@Override public String mojoVersion() {
return "1.40";
}
}