All Downloads are FREE. Search and download functionalities are using the official Maven repository.

hex.schemas.Word2VecTransformV3 Maven / Gradle / Ivy

package hex.schemas;

import hex.word2vec.Word2VecModel;
import water.Iced;
import water.api.API;
import water.api.schemas3.KeyV3;
import water.api.schemas3.SchemaV3;

public class Word2VecTransformV3 extends SchemaV3 {
  @API(help="Source word2vec Model", required = true, direction = API.Direction.INPUT)
  public KeyV3.ModelKeyV3 model;
  @API(help = "Words Frame", required = true, direction = API.Direction.INPUT)
  public KeyV3.FrameKeyV3 words_frame;
  @API(help="Method of aggregating word-vector sequences into a single vector", values = {"NONE", "AVERAGE"},
          direction = API.Direction.INPUT)
  public Word2VecModel.AggregateMethod aggregate_method;
  @API(help = "Word Vectors Frame", direction = API.Direction.OUTPUT)
  public KeyV3.FrameKeyV3 vectors_frame;
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy