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)
Saveable
This 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()