Interface NaiveBayesParams

All Superinterfaces:
HasFeaturesCol, HasLabelCol, HasPredictionCol, HasWeightCol, Identifiable, Params, PredictorParams, Serializable, scala.Serializable
All Known Implementing Classes:
NaiveBayes, NaiveBayesModel

public interface NaiveBayesParams extends PredictorParams, HasWeightCol
Params for Naive Bayes Classifiers.
  • Method Details

    • getModelType

      String getModelType()
    • getSmoothing

      double getSmoothing()
    • modelType

      Param<String> modelType()
      The model type which is a string (case-sensitive). Supported options: "multinomial", "complement", "bernoulli", "gaussian". (default = multinomial)
    • smoothing

      DoubleParam smoothing()
      The smoothing parameter. (default = 1.0).