public class CountVectorizer extends Estimator<CountVectorizerModel> implements DefaultParamsWritable
CountVectorizerModel
.Constructor and Description |
---|
CountVectorizer() |
CountVectorizer(String uid) |
Modifier and Type | Method and Description |
---|---|
static BooleanParam |
binary() |
BooleanParam |
binary()
Binary toggle to control the output vector values.
|
static Params |
clear(Param<?> param) |
CountVectorizer |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
static String |
explainParam(Param<?> param) |
static String |
explainParams() |
static ParamMap |
extractParamMap() |
static ParamMap |
extractParamMap(ParamMap extra) |
CountVectorizerModel |
fit(Dataset<?> dataset)
Fits a model to the input data.
|
static <T> scala.Option<T> |
get(Param<T> param) |
static boolean |
getBinary() |
boolean |
getBinary() |
static <T> scala.Option<T> |
getDefault(Param<T> param) |
static String |
getInputCol() |
static double |
getMaxDF() |
double |
getMaxDF() |
static double |
getMinDF() |
double |
getMinDF() |
static double |
getMinTF() |
double |
getMinTF() |
static <T> T |
getOrDefault(Param<T> param) |
static String |
getOutputCol() |
static Param<Object> |
getParam(String paramName) |
static int |
getVocabSize() |
int |
getVocabSize() |
static <T> boolean |
hasDefault(Param<T> param) |
static boolean |
hasParam(String paramName) |
static Param<String> |
inputCol() |
static boolean |
isDefined(Param<?> param) |
static boolean |
isSet(Param<?> param) |
static CountVectorizer |
load(String path) |
static DoubleParam |
maxDF() |
DoubleParam |
maxDF()
Specifies the maximum number of different documents a term could appear in to be included
in the vocabulary.
|
static DoubleParam |
minDF() |
DoubleParam |
minDF()
Specifies the minimum number of different documents a term must appear in to be included
in the vocabulary.
|
static DoubleParam |
minTF() |
DoubleParam |
minTF()
Filter to ignore rare words in a document.
|
static Param<String> |
outputCol() |
static Param<?>[] |
params() |
static void |
save(String path) |
static <T> Params |
set(Param<T> param,
T value) |
CountVectorizer |
setBinary(boolean value) |
CountVectorizer |
setInputCol(String value) |
CountVectorizer |
setMaxDF(double value) |
CountVectorizer |
setMinDF(double value) |
CountVectorizer |
setMinTF(double value) |
CountVectorizer |
setOutputCol(String value) |
CountVectorizer |
setVocabSize(int value) |
static String |
toString() |
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
StructType |
validateAndTransformSchema(StructType schema)
Validates and transforms the input schema.
|
static IntParam |
vocabSize() |
IntParam |
vocabSize()
Max size of the vocabulary.
|
static MLWriter |
write() |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getInputCol, inputCol
getOutputCol, outputCol
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
toString
write
save
initializeLogging, initializeLogIfNecessary, initializeLogIfNecessary, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public CountVectorizer(String uid)
public CountVectorizer()
public static CountVectorizer load(String path)
public static String toString()
public static Param<?>[] params()
public static String explainParam(Param<?> param)
public static String explainParams()
public static final boolean isSet(Param<?> param)
public static final boolean isDefined(Param<?> param)
public static boolean hasParam(String paramName)
public static Param<Object> getParam(String paramName)
public static final <T> scala.Option<T> get(Param<T> param)
public static final <T> T getOrDefault(Param<T> param)
public static final <T> scala.Option<T> getDefault(Param<T> param)
public static final <T> boolean hasDefault(Param<T> param)
public static final ParamMap extractParamMap()
public static final Param<String> inputCol()
public static final String getInputCol()
public static final Param<String> outputCol()
public static final String getOutputCol()
public static IntParam vocabSize()
public static int getVocabSize()
public static DoubleParam minDF()
public static double getMinDF()
public static DoubleParam maxDF()
public static double getMaxDF()
public static DoubleParam minTF()
public static double getMinTF()
public static BooleanParam binary()
public static boolean getBinary()
public static void save(String path) throws java.io.IOException
java.io.IOException
public static MLWriter write()
public String uid()
Identifiable
uid
in interface Identifiable
public CountVectorizer setInputCol(String value)
public CountVectorizer setOutputCol(String value)
public CountVectorizer setVocabSize(int value)
public CountVectorizer setMinDF(double value)
public CountVectorizer setMaxDF(double value)
public CountVectorizer setMinTF(double value)
public CountVectorizer setBinary(boolean value)
public CountVectorizerModel fit(Dataset<?> dataset)
Estimator
fit
in class Estimator<CountVectorizerModel>
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Check transform validity and derive the output schema from the input schema.
We check validity for interactions between parameters during transformSchema
and
raise an exception if any parameter value is invalid. Parameter value checks which
do not depend on other parameters are handled by Param.validate()
.
Typical implementation should first conduct verification on schema change and parameter validity, including complex parameter interaction checks.
transformSchema
in class PipelineStage
schema
- (undocumented)public CountVectorizer copy(ParamMap extra)
Params
defaultCopy()
.copy
in interface Params
copy
in class Estimator<CountVectorizerModel>
extra
- (undocumented)public BooleanParam binary()
public boolean getBinary()
public double getMaxDF()
public double getMinDF()
public double getMinTF()
public int getVocabSize()
public DoubleParam maxDF()
Default: (2^63^) - 1
public DoubleParam minDF()
Default: 1.0
public DoubleParam minTF()
Note that the parameter is only used in transform of CountVectorizerModel
and does not
affect fitting.
Default: 1.0
public StructType validateAndTransformSchema(StructType schema)
public IntParam vocabSize()
Default: 2^18^