|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
Object org.apache.spark.ml.PipelineStage org.apache.spark.ml.Transformer org.apache.spark.ml.Model<Word2VecModel> org.apache.spark.ml.feature.Word2VecModel
public class Word2VecModel
:: Experimental ::
Model fitted by Word2Vec
.
Method Summary | |
---|---|
Word2VecModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params. |
int |
getMinCount()
|
int |
getNumPartitions()
|
int |
getVectorSize()
|
IntParam |
minCount()
The minimum number of times a token must appear to be included in the word2vec model's vocabulary. |
IntParam |
numPartitions()
Number of partitions for sentences of words. |
Word2VecModel |
setInputCol(String value)
|
Word2VecModel |
setOutputCol(String value)
|
DataFrame |
transform(DataFrame dataset)
Transform a sentence column to a vector column to represent the whole sentence. |
StructType |
transformSchema(StructType schema)
:: DeveloperApi :: |
String |
uid()
|
StructType |
validateAndTransformSchema(StructType schema)
Validate and transform the input schema. |
IntParam |
vectorSize()
The dimension of the code that you want to transform from words. |
Methods inherited from class org.apache.spark.ml.Model |
---|
hasParent, parent, setParent |
Methods inherited from class org.apache.spark.ml.Transformer |
---|
transform, transform, transform |
Methods inherited from class Object |
---|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
Methods inherited from interface org.apache.spark.ml.param.Params |
---|
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams |
Methods inherited from interface org.apache.spark.Logging |
---|
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning |
Method Detail |
---|
public String uid()
public Word2VecModel setInputCol(String value)
public Word2VecModel setOutputCol(String value)
public DataFrame transform(DataFrame dataset)
transform
in class Transformer
dataset
- (undocumented)
public StructType transformSchema(StructType schema)
PipelineStage
Derives the output schema from the input schema.
transformSchema
in class PipelineStage
schema
- (undocumented)
public Word2VecModel copy(ParamMap extra)
Params
copy
in interface Params
copy
in class Model<Word2VecModel>
extra
- (undocumented)
defaultCopy()
public IntParam vectorSize()
public int getVectorSize()
public IntParam numPartitions()
public int getNumPartitions()
public IntParam minCount()
public int getMinCount()
public StructType validateAndTransformSchema(StructType schema)
schema
- (undocumented)
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |