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;
}
Aggregations