Class: Word2VecModel

eclairjs/ml/feature.Word2VecModel

Model fitted by Word2Vec.

Constructor

new Word2VecModel()

Source:

Extends

Methods

(static) load(path) → {Word2VecModel}

Parameters:
Name Type Description
path string
Source:
Returns:
Type
Word2VecModel

(static) read() → {MLReader}

Source:
Returns:
Type
MLReader

copy(extra) → {module:eclairjs/ml/feature.Word2VecModel}

Parameters:
Name Type Description
extra module:eclairjs/ml/param.ParamMap
Source:
Returns:
Type
module:eclairjs/ml/feature.Word2VecModel

findSynonyms(word, num) → {module:eclairjs/sql.Dataset}

Find "num" number of words closest in similarity to the given word. Returns a Dataset with the words and the cosine similarities between the synonyms and the given word.
Parameters:
Name Type Description
word string
num number
Source:
Returns:
Type
module:eclairjs/sql.Dataset

save(path) → {Promise.<Void>}

Saves this ML instance to the input path, a shortcut of `write.save(path)`.
Parameters:
Name Type Description
path string
Inherited From:
Source:
Returns:
A Promise that resolves to nothing.
Type
Promise.<Void>

setInputCol(value) → {module:eclairjs/ml/feature.Word2VecModel}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/feature.Word2VecModel

setOutputCol(value) → {module:eclairjs/ml/feature.Word2VecModel}

Parameters:
Name Type Description
value string
Source:
Returns:
Type
module:eclairjs/ml/feature.Word2VecModel

transform(dataset) → {module:eclairjs/sql.Dataset}

Transform a sentence column to a vector column to represent the whole sentence. The transform is performed by averaging all word vectors it contains.
Parameters:
Name Type Description
dataset module:eclairjs/sql.Dataset
Source:
Returns:
Type
module:eclairjs/sql.Dataset

transformSchema(schema) → {module:eclairjs/sql/types.StructType}

Parameters:
Name Type Description
schema module:eclairjs/sql/types.StructType
Source:
Returns:
Type
module:eclairjs/sql/types.StructType

write() → {MLWriter}

Overrides:
Source:
Returns:
Type
MLWriter