public class NaiveBayes extends Object implements Estimator<NaiveBayes,NaiveBayesModel>, NaiveBayesParams<NaiveBayes>
See https://en.wikipedia.org/wiki/Naive_Bayes_classifier.
SMOOTHING
MODEL_TYPE
FEATURES_COL
PREDICTION_COL
LABEL_COL
Constructor and Description |
---|
NaiveBayes() |
Modifier and Type | Method and Description |
---|---|
NaiveBayesModel |
fit(org.apache.flink.table.api.Table... inputs)
Trains on the given inputs and produces a Model.
|
Map<Param<?>,Object> |
getParamMap()
Returns a map which should contain value for every parameter that meets one of the following
conditions.
|
static NaiveBayes |
load(org.apache.flink.streaming.api.environment.StreamExecutionEnvironment env,
String path) |
void |
save(String path)
Saves this stage to the given path.
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getSmoothing, setSmoothing
getModelType, setModelType
getFeaturesCol, setFeaturesCol
getPredictionCol, setPredictionCol
get, getParam, set
getLabelCol, setLabelCol
public NaiveBayesModel fit(org.apache.flink.table.api.Table... inputs)
Estimator
fit
in interface Estimator<NaiveBayes,NaiveBayesModel>
inputs
- a list of tablespublic void save(String path) throws IOException
Stage
save
in interface Stage<NaiveBayes>
IOException
public static NaiveBayes load(org.apache.flink.streaming.api.environment.StreamExecutionEnvironment env, String path) throws IOException
IOException
public Map<Param<?>,Object> getParamMap()
WithParams
1) set(...) has been called to set value for this parameter.
2) The parameter is a public final field of this WithParams instance. This includes fields inherited from its interfaces and super-classes.
The subclass which implements this interface could meet this requirement by returning a
member field of the given map type, after having initialized this member field using the
ParamUtils.initializeMapWithDefaultValues(Map, WithParams)
method.
getParamMap
in interface WithParams<NaiveBayes>
Copyright © 2019–2023 The Apache Software Foundation. All rights reserved.