hex.schemas.SynonymV1 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.Key;
import water.api.API;
import water.api.Schema;
import hex.schemas.SynonymsHandler.Synonyms;

public class SynonymV1 extends Schema {

  // Input fields
  @API(help="A word2vec model key.") public Key key;
  @API(help="The target string to find synonyms.") public String target;
  @API(help="Find the top `cnt` synonyms of the target word.") public int cnt;


  //Outputs
  @API(help="The synonyms.", direction=API.Direction.OUTPUT) public String[] synonyms;
  @API(help="The cosine similarities.", direction=API.Direction.OUTPUT) public float[]  cos_sim;

/*  @Override public SynonymsHandler.Synonyms fillImpl(Synonyms c) {
    //c._w2vec_key = Key.make(key);
    c._cnt = cnt;
    c._target = target;
    return c;
  }
  */
}