hex.tree.xgboost.exec.XGBoostExecReq Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-ext-xgboost Show documentation

H2O XGBoost support

There is a newer version: 3.46.0.6

package hex.tree.xgboost.exec;

import water.BootstrapFreezable;
import water.Iced;
import water.util.IcedHashMapGeneric;

import java.util.Arrays;
import java.util.Map;

public class XGBoostExecReq extends Iced implements BootstrapFreezable {

    @Override
    public String toString() {
        return "XGBoostExecReq{}";
    }

    public static class Init extends XGBoostExecReq {
        public int num_nodes;
        public IcedHashMapGeneric.IcedHashMapStringObject parms;
        public String save_matrix_path;
        public String[] nodes;
        public boolean has_checkpoint;

        public void setParms(Map parms) {
            this.parms = new IcedHashMapGeneric.IcedHashMapStringObject();
            this.parms.putAll(parms);
        }

        @Override
        public String toString() {
            return "XGBoostExecReq.Init{" +
                "num_nodes=" + num_nodes +
                ", parms=" + parms +
                ", save_matrix_path='" + save_matrix_path + '\'' +
                ", nodes=" + Arrays.toString(nodes) +
                ", has_checkpoint=" + has_checkpoint +
                '}';
        }
    }

    public static class Update extends XGBoostExecReq {
        public int treeId;

        @Override
        public String toString() {
            return "XGBoostExecReq.Update{" +
                "treeId=" + treeId +
                '}';
        }
    }

    public static class GetEvalMetric extends XGBoostExecReq {
        @Override
        public String toString() {
            return "XGBoostExecReq.GetEvalMetric{}";
        }
    }

}