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