hex.schemas.Word2VecModelV3 Maven / Gradle / Ivy

Go to download
package hex.schemas;

import hex.word2vec.Word2VecModel;
import water.api.schemas3.ModelOutputSchemaV3;
import water.api.schemas3.ModelSchemaV3;
import water.api.*;


public class Word2VecModelV3 extends ModelSchemaV3 {

  public static final class Word2VecModelOutputV3 extends ModelOutputSchemaV3 {
    @API(help = "Number of epochs executed")
    public int epochs;
  }


  public Word2VecV3.Word2VecParametersV3 createParametersSchema() { return new Word2VecV3.Word2VecParametersV3(); }
  public Word2VecModelOutputV3 createOutputSchema() { return new Word2VecModelOutputV3(); }

  // Version&Schema-specific filling into the impl
  @Override public Word2VecModel createImpl() {
    Word2VecModel.Word2VecParameters parms = parameters.createImpl();
    return new Word2VecModel( model_id.key(), parms, null);
  }
}