hex.schemas.CoxPHModelV2 Maven / Gradle / Ivy

Go to download
package hex.schemas;

import hex.coxph.CoxPHModel;
import water.Key;
import water.api.ModelOutputSchema;
import water.api.ModelSchema;
import water.util.PojoUtils;

public class CoxPHModelV2 extends ModelSchema {

  public static final class CoxPHModelOutputV2 extends ModelOutputSchema {
    //FIXME
    //add output fields

    @Override public CoxPHModel.CoxPHOutput createImpl() {
      CoxPHModel.CoxPHOutput impl = new CoxPHModel.CoxPHOutput(null);
      PojoUtils.copyProperties(impl, this, PojoUtils.FieldNaming.DEST_HAS_UNDERSCORES);
      return impl;
    }

    // Version&Schema-specific filling from the handler
    @Override public CoxPHModelOutputV2 fillFromImpl( CoxPHModel.CoxPHOutput impl) {
      PojoUtils.copyProperties(this, impl, PojoUtils.FieldNaming.ORIGIN_HAS_UNDERSCORES);
      return this;
    }


  } // CoxPHModelOutputV2

  // TOOD: I think we can implement the following two in ModelSchema, using reflection on the type parameters.
  public CoxPHV2.CoxPHParametersV2 createParametersSchema() { return new CoxPHV2.CoxPHParametersV2(); }
  public CoxPHModelOutputV2 createOutputSchema() { return new CoxPHModelOutputV2(); }

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

  // Version&Schema-specific filling into the impl
  @Override public CoxPHModel createImpl() {
    CoxPHV2.CoxPHParametersV2 p = ((CoxPHV2.CoxPHParametersV2)this.parameters);
    CoxPHModel.CoxPHParameters parms = p.createImpl();
    return new CoxPHModel(Key.make() /*dest*/, parms, new CoxPHModel.CoxPHOutput(null));
  }

  // Version&Schema-specific filling from the impl
  @Override public CoxPHModelV2 fillFromImpl( CoxPHModel kmm ) {
    return super.fillFromImpl(kmm);
  }
}