public class Word2VecModel extends Object implements scala.Serializable, Saveable
| Constructor and Description |
|---|
Word2VecModel(scala.collection.immutable.Map<String,float[]> model) |
| Modifier and Type | Method and Description |
|---|---|
scala.Tuple2<String,Object>[] |
findSynonyms(String word,
int num) |
scala.Tuple2<String,Object>[] |
findSynonyms(Vector vector,
int num) |
scala.collection.immutable.Map<String,float[]> |
getVectors() |
static Word2VecModel |
load(SparkContext sc,
String path) |
void |
save(SparkContext sc,
String path)
Save this model to the given path.
|
Vector |
transform(String word) |
scala.collection.immutable.Map<String,Object> |
wordIndex() |
float[] |
wordVectors() |
public Word2VecModel(scala.collection.immutable.Map<String,float[]> model)
public static Word2VecModel load(SparkContext sc, String path)
public scala.collection.immutable.Map<String,Object> wordIndex()
public float[] wordVectors()
public void save(SparkContext sc, String path)
SaveableThis saves: - human-readable (JSON) model metadata to path/metadata/ - Parquet formatted data to path/data/
The model may be loaded using Loader.load.
public Vector transform(String word)
public scala.Tuple2<String,Object>[] findSynonyms(String word,
int num)
public scala.Tuple2<String,Object>[] findSynonyms(Vector vector, int num)
public scala.collection.immutable.Map<String,float[]> getVectors()