org.jpmml.sparkml.model

Class NaiveBayesModelConverter

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

org.jpmml.sparkml.ClassificationModelConverter<org.apache.spark.ml.classification.NaiveBayesModel>

org.jpmml.sparkml.model.NaiveBayesModelConverter

All Implemented Interfaces:

org.jpmml.converter.HasOptions, HasSparkMLOptions, HasRegressionTableOptions

public class NaiveBayesModelConverter
extends ClassificationModelConverter<org.apache.spark.ml.classification.NaiveBayesModel>
implements HasRegressionTableOptions

Field Summary

Fields inherited from interface org.jpmml.sparkml.model.HasRegressionTableOptions

OPTION_LOOKUP_THRESHOLD

Constructor Summary

Constructors

Constructor and Description
NaiveBayesModelConverter(org.apache.spark.ml.classification.NaiveBayesModel model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.regression.RegressionModel	encodeModel(org.jpmml.converter.Schema schema)

Methods inherited from class org.jpmml.sparkml.ClassificationModelConverter

getMiningFunction, registerOutputFields

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeSchema, hasLabelCol, hasPredictionCol, hasProbabilityCol, registerModel

Methods inherited from class org.jpmml.sparkml.TransformerConverter

getOption, getOptions, getTransformer, setOptions

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

NaiveBayesModelConverter

public NaiveBayesModelConverter(org.apache.spark.ml.classification.NaiveBayesModel model)

Method Detail

encodeModel

public org.dmg.pmml.regression.RegressionModel encodeModel(org.jpmml.converter.Schema schema)

Specified by:

encodeModel in class ModelConverter<org.apache.spark.ml.classification.NaiveBayesModel>