hex.schemas.DeepLearningHandler Maven / Gradle / Ivy

Go to download

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

H2O Algorithms

There is a newer version: 3.46.0.6

package hex.schemas;

import hex.deeplearning.DeepLearning;
import hex.deeplearning.DeepLearningModel;
import water.H2O;
import water.api.Handler;

/** TODO: only used by old-school web ui: remove!  ModelBuilderHander does this for all the algos.  */
@Deprecated
public class DeepLearningHandler extends Handler {
  @Override protected int min_ver() { return 2; }
  @Override protected int max_ver() { return Integer.MAX_VALUE; }

  public DeepLearningHandler() {}

  // TODO: move this into a new ModelBuilderHandler superclass
  // TODO: also add a score method in the new ModelBuilderHandler superclass
  public DeepLearningV2 train(int version, DeepLearning builder) {
    DeepLearningModel.DeepLearningParameters parms = builder._parms;
    assert parms != null; /* impl._job = */
    builder.trainModel();
    DeepLearningV2 schema = schema(version); // TODO: superclass!
    schema.parameters = new DeepLearningV2.DeepLearningParametersV2();
    schema.job = builder._key;
    return schema;
  }
  @Override protected DeepLearningV2 schema(int version) { DeepLearningV2 schema = new DeepLearningV2(); schema.parameters = schema.createParametersSchema(); return schema; }
  @Override public void compute2() { throw H2O.fail(); }
}