hex.tree.drf.DrfMojoWriter 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.tree.drf;

import hex.tree.SharedTreeMojoWriter;

import java.io.IOException;

/**
 * Mojo definition for DRF model.
 */
public class DrfMojoWriter extends SharedTreeMojoWriter {

  @SuppressWarnings("unused")  // Called through reflection in ModelBuildersHandler
  public DrfMojoWriter() {}

  public DrfMojoWriter(DRFModel model) { super(model); }

  @Override public String mojoVersion() {
    return "1.40";
  }

  @Override
  protected void writeModelData() throws IOException {
    super.writeModelData();
    writekv("binomial_double_trees", model._parms._binomial_double_trees);
  }
}