org.maochen.nlp.ml.classifier.naivebayes

Class NaiveBayesClassifier

java.lang.Object

org.maochen.nlp.ml.classifier.naivebayes.NaiveBayesClassifier

All Implemented Interfaces:

org.maochen.nlp.ml.IClassifier

public class NaiveBayesClassifier
extends Object
implements org.maochen.nlp.ml.IClassifier

Created by Maochen on 12/3/14.

Constructor Summary

Constructors

Constructor and Description
NaiveBayesClassifier()
NaiveBayesClassifier(InputStream modelInputStream)

Method Summary

Modifier and Type	Method and Description
void	loadModel(InputStream inputStream)
void	persistModel(String filename)
Map<String,Double>	predict(org.maochen.nlp.ml.Tuple predict)
String	predictLabel(org.maochen.nlp.ml.Tuple predict)
static List<org.maochen.nlp.ml.Tuple>	readTrainingData(String filename, String delimiter)
void	setParameter(Properties props)
static void	splitData(String originalTrainingDataFile)
org.maochen.nlp.ml.IClassifier	train(List<org.maochen.nlp.ml.Tuple> trainingData)
static void	writeToFile(List<org.maochen.nlp.ml.Tuple> dataset, String filename)

Methods inherited from class java.lang.Object

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

Constructor Detail

NaiveBayesClassifier

public NaiveBayesClassifier(InputStream modelInputStream)

NaiveBayesClassifier

public NaiveBayesClassifier()

Method Detail

setParameter

public void setParameter(Properties props)

Specified by:

setParameter in interface org.maochen.nlp.ml.IClassifier

predict

public Map<String,Double> predict(org.maochen.nlp.ml.Tuple predict)

Specified by:

predict in interface org.maochen.nlp.ml.IClassifier

predictLabel

public String predictLabel(org.maochen.nlp.ml.Tuple predict)

train

public org.maochen.nlp.ml.IClassifier train(List<org.maochen.nlp.ml.Tuple> trainingData)

Specified by:

train in interface org.maochen.nlp.ml.IClassifier

persistModel

public void persistModel(String filename)

Specified by:

persistModel in interface org.maochen.nlp.ml.IClassifier

loadModel

public void loadModel(InputStream inputStream)

Specified by:

loadModel in interface org.maochen.nlp.ml.IClassifier

readTrainingData

public static List<org.maochen.nlp.ml.Tuple> readTrainingData(String filename,
                                                              String delimiter)

writeToFile

public static void writeToFile(List<org.maochen.nlp.ml.Tuple> dataset,
                               String filename)

splitData

public static void splitData(String originalTrainingDataFile)