hex.api.DeepLearningBuilderHandler 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.api;

import hex.deeplearning.DeepLearning;
import hex.schemas.DeepLearningV2;
import water.H2O;
import water.api.ModelBuilderHandler;

public class DeepLearningBuilderHandler extends ModelBuilderHandler {

  @Override protected DeepLearningV2 schema(int version) {
    switch (version) {
      case 2:   { DeepLearningV2 b = new DeepLearningV2(); b.parameters = b.createParametersSchema(); return b; }
      default:  throw H2O.fail("Bad version for ModelBuilder schema: " + version);
    }
  }
}