hex.schemas.Word2VecSynonymsV3 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.schemas;

import water.Iced;
import water.api.API;
import water.api.schemas3.KeyV3;
import water.api.schemas3.SchemaV3;

public class Word2VecSynonymsV3 extends SchemaV3 {
  @API(help="Source word2vec Model", required = true, direction = API.Direction.INPUT)
  public KeyV3.ModelKeyV3 model;
  @API(help="Target word to find synonyms for", required = true, direction = API.Direction.INPUT)
  public String word;
  @API(help="Number of synonyms", required = true, direction = API.Direction.INPUT)
  public int count;
  @API(help="Synonymous words")
  public String[] synonyms;
  @API(help="Similarity scores")
  public double[] scores;
}