public class NaiveBayesModel extends ProbabilisticClassificationModel<Vector,NaiveBayesModel> implements NaiveBayesParams, MLWritable
NaiveBayes
param: pi log of class priors, whose dimension is C (number of classes)
param: theta log of class conditional probabilities, whose dimension is C (number of classes)
by D (number of features)| Modifier and Type | Method and Description |
|---|---|
NaiveBayesModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
static NaiveBayesModel |
load(String path) |
int |
numClasses()
Number of classes (values which the label can take).
|
int |
numFeatures()
Returns the number of features the model was trained on.
|
Vector |
pi() |
static MLReader<NaiveBayesModel> |
read() |
Matrix |
theta() |
String |
toString() |
String |
uid()
An immutable unique ID for the object and its derivatives.
|
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
normalizeToProbabilitiesInPlace, setProbabilityCol, setThresholds, transformpredict, setRawPredictionColsetFeaturesCol, setPredictionCol, transformSchematransform, transform, transformgetModelType, getSmoothing, modelType, smoothingvalidateAndTransformSchemagetLabelCol, labelColfeaturesCol, getFeaturesColgetPredictionCol, predictionColclear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwngetWeightCol, weightColsavevalidateAndTransformSchemagetRawPredictionCol, rawPredictionColgetProbabilityCol, probabilityColgetThresholds, thresholdsinitializeLogging, initializeLogIfNecessary, initializeLogIfNecessary, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarningpublic static MLReader<NaiveBayesModel> read()
public static NaiveBayesModel load(String path)
public String uid()
Identifiableuid in interface Identifiablepublic Vector pi()
public Matrix theta()
public int numFeatures()
PredictionModelnumFeatures in class PredictionModel<Vector,NaiveBayesModel>public int numClasses()
ClassificationModelnumClasses in class ClassificationModel<Vector,NaiveBayesModel>public NaiveBayesModel copy(ParamMap extra)
ParamsdefaultCopy().copy in interface Paramscopy in class Model<NaiveBayesModel>extra - (undocumented)public String toString()
toString in interface IdentifiabletoString in class Objectpublic MLWriter write()
MLWritableMLWriter instance for this ML instance.write in interface MLWritable