Arguments
This function internally uses scikit-learn functions MultinomialNB, BernoulliNB, GaussianNB through teradataml Open source ML functions.
Transformed data won’t have features, rawPrediction, probability columns.
| PySpark Argument | Open Source Function Argument Name | Notes |
|---|---|---|
| predictionCol | Not yet available. | |
| probabilityCol | Not yet available. | |
| rawPredictionCol | Not yet available. | |
| smoothing | alpha | |
| modelType | scikit-learn function MultinomialNB, BernoulliNB, GaussianNB. | |
| thresholds | Not yet available. | |
| weightCol | Not yet available. |
Attributes/Methods
| Attribute/Method Name | Supported | Notes |
|---|---|---|
| clear | ||
| copy | ||
| explainParam | ||
| explainParams | ||
| extractParamMap | ||
| fit | ||
| fitMultiple | ||
| getFeaturesCol | ||
| getLabelCol | ||
| getModelType | ||
| getOrDefault | ||
| getParam | ||
| getPredictionCol | ||
| getProbabilityCol | ||
| getRawPredictionCol | ||
| getSmoothing | ||
| getThresholds | ||
| getWeightCol | ||
| hasDefault | ||
| hasParam | ||
| isDefined | ||
| isSet | ||
| load | ||
| read | ||
| save | ||
| set | ||
| setFeaturesCol | ||
| setLabelCol | ||
| setModelType | ||
| setParams | ||
| setPredictionCol | ||
| setProbabilityCol | ||
| setRawPredictionCol | ||
| setSmoothing | ||
| setThresholds | ||
| setWeightCol | ||
| write | ||
| featuresCol | ||
| labelCol | ||
| modelType | ||
| params | ||
| predictionCol | ||
| probabilityCol | ||
| rawPredictionCol | ||
| smoothing | ||
| thresholds | ||
| weightCol |