Search in sources :

Example 1 with ModelsInfoV4

use of water.api.schemas4.ModelsInfoV4 in project h2o-3 by h2oai.

the class ModelBuildersHandler method modelsInfo.

// called through reflection by RequestServer
@SuppressWarnings("unused")
public ModelsInfoV4 modelsInfo(int version, ListRequestV4 m) {
    String[] algos = ModelBuilder.algos();
    ModelInfoV4[] infos = new ModelInfoV4[algos.length];
    ModelsInfoV4 res = new ModelsInfoV4();
    for (int i = 0; i < algos.length; i++) {
        ModelBuilder builder = ModelBuilder.make(algos[i], null, null);
        infos[i] = new ModelInfoV4();
        infos[i].algo = algos[i];
        infos[i].maturity = builder.builderVisibility() == ModelBuilder.BuilderVisibility.Stable ? "stable" : builder.builderVisibility() == ModelBuilder.BuilderVisibility.Beta ? "beta" : "alpha";
        infos[i].have_mojo = builder.haveMojo();
        infos[i].have_pojo = builder.havePojo();
        infos[i].mojo_version = infos[i].have_mojo ? detectMojoVersion(builder) : null;
    }
    res.models = infos;
    return res;
}
Also used : ModelInfoV4(water.api.schemas4.ModelInfoV4) ModelBuilder(hex.ModelBuilder) ModelsInfoV4(water.api.schemas4.ModelsInfoV4)

Aggregations

ModelBuilder (hex.ModelBuilder)1 ModelInfoV4 (water.api.schemas4.ModelInfoV4)1 ModelsInfoV4 (water.api.schemas4.ModelsInfoV4)1