ae.teletronics.nlp.entityextraction.types.sender

Class NaiveBayesModel

java.lang.Object

ae.teletronics.nlp.entityextraction.types.sender.NaiveBayesModel

All Implemented Interfaces:

Model

public class NaiveBayesModel
extends Object
implements Model

Created by hhravn on 06/06/16.

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.apache.spark.mllib.classification.NaiveBayesModel	model()
org.apache.spark.rdd.RDD<Object>	predict(org.apache.spark.rdd.RDD<org.apache.spark.mllib.linalg.Vector> data)
void	save(org.apache.spark.SparkContext sc, String fileName)
org.apache.spark.rdd.RDD<scala.Tuple2<Object,Object>>	test(org.apache.spark.rdd.RDD<org.apache.spark.mllib.regression.LabeledPoint> data)

Methods inherited from class java.lang.Object

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

Constructor Detail

NaiveBayesModel

public NaiveBayesModel(org.apache.spark.mllib.classification.NaiveBayesModel model)

Method Detail

model

public org.apache.spark.mllib.classification.NaiveBayesModel model()

predict

public org.apache.spark.rdd.RDD<Object> predict(org.apache.spark.rdd.RDD<org.apache.spark.mllib.linalg.Vector> data)

Specified by:

predict in interface Model

test

public org.apache.spark.rdd.RDD<scala.Tuple2<Object,Object>> test(org.apache.spark.rdd.RDD<org.apache.spark.mllib.regression.LabeledPoint> data)

Specified by:

test in interface Model

save

public void save(org.apache.spark.SparkContext sc,
                 String fileName)

Specified by:

save in interface Model