Search in sources :

Example 6 with ModelBuilder

use of hex.ModelBuilder in project h2o-3 by h2oai.

the class RegisterAlgos method register.

// Register the algorithms and their builder handlers:
@Override
public void register(String relativeResourcePath) throws ClassNotFoundException {
    // List of algorithms
    ModelBuilder[] algos = new ModelBuilder[] { new hex.deeplearning.DeepLearning(true), new hex.glm.GLM(true), new hex.glrm.GLRM(true), new hex.kmeans.KMeans(true), new hex.naivebayes.NaiveBayes(true), new hex.pca.PCA(true), new hex.svd.SVD(true), new hex.tree.drf.DRF(true), new hex.tree.gbm.GBM(true), new hex.aggregator.Aggregator(true), new hex.deepwater.DeepWater(true), new hex.word2vec.Word2Vec(true), new hex.ensemble.StackedEnsemble(true) };
    // "Word2Vec", "Example", "Grep"
    for (ModelBuilder algo : algos) {
        String base = algo.getClass().getSimpleName();
        String lbase = base.toLowerCase();
        Class<? extends water.api.Handler> bh_clz = water.api.ModelBuilderHandler.class;
        int version = 3;
        // SVD is experimental still
        if (base.equals("SVD"))
            version = 99;
        // Aggregator is experimental still
        if (base.equals("Aggregator"))
            version = 99;
        // StackedEnsemble is experimental still
        if (base.equals("StackedEnsemble"))
            version = 99;
        H2O.register("POST /" + version + "/ModelBuilders/" + lbase, bh_clz, "train", "train_" + lbase, "Train a " + base + " model.");
        H2O.register("POST /" + version + "/ModelBuilders/" + lbase + "/parameters", bh_clz, "validate_parameters", "validate_" + lbase, "Validate a set of " + base + " model builder parameters.");
        // Grid search is experimental feature
        H2O.register("POST /99/Grid/" + lbase, GridSearchHandler.class, "train", "grid_search_" + lbase, "Run grid search for " + base + " model.");
    }
    H2O.register("POST /3/MakeGLMModel", MakeGLMModelHandler.class, "make_model", "make_glm_model", "Make a new GLM model based on existing one");
    H2O.register("GET /3/GetGLMRegPath", MakeGLMModelHandler.class, "extractRegularizationPath", "glm_regularization_path", "Get full regularization path");
    H2O.register("GET /3/ComputeGram", MakeGLMModelHandler.class, "computeGram", "weighted_gram_matrix", "Get weighted gram matrix");
    H2O.register("GET /3/Word2VecSynonyms", Word2VecHandler.class, "findSynonyms", "word2vec_synonyms", "Find synonyms using a word2vec model");
    H2O.register("GET /3/Word2VecTransform", Word2VecHandler.class, "transform", "word2vec_transform", "Transform words to vectors using a word2vec model");
    H2O.register("POST /3/DataInfoFrame", MakeGLMModelHandler.class, "getDataInfoFrame", "glm_datainfo_frame", "Test only");
}
Also used : ModelBuilder(hex.ModelBuilder)

Aggregations

ModelBuilder (hex.ModelBuilder)6 ModelBuilderSchema (hex.schemas.ModelBuilderSchema)2 Grid (hex.grid.Grid)1 JobV3 (water.api.schemas3.JobV3)1 ModelInfoV4 (water.api.schemas4.ModelInfoV4)1 ModelsInfoV4 (water.api.schemas4.ModelsInfoV4)1