hex.schemas.DRFModelV3 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.tree.drf.DRFModel;

public class DRFModelV3 extends SharedTreeModelV3 {

  public static final class DRFModelOutputV3 extends SharedTreeModelV3.SharedTreeModelOutputV3 {}

  public DRFV3.DRFParametersV3 createParametersSchema() { return new DRFV3.DRFParametersV3(); }
  public DRFModelOutputV3 createOutputSchema() { return new DRFModelOutputV3(); }

  //==========================
  // Custom adapters go here

  // Version&Schema-specific filling into the impl
  @Override public DRFModel createImpl() {
    DRFV3.DRFParametersV3 p = this.parameters;
    DRFModel.DRFParameters parms = p.createImpl();
    return new DRFModel( model_id.key(), parms, new DRFModel.DRFOutput(null) );
  }
}