public class NaiveBayesClassifier extends Object implements IClassifier
| Constructor and Description |
|---|
NaiveBayesClassifier() |
NaiveBayesClassifier(InputStream modelInputStream) |
| Modifier and Type | Method and Description |
|---|---|
void |
loadModel(InputStream inputStream) |
void |
persistModel(String filename) |
Map<String,Double> |
predict(Tuple predict) |
String |
predictLabel(Tuple predict) |
static List<Tuple> |
readTrainingData(String filename,
String delimiter) |
void |
setParameter(Properties props) |
static void |
splitData(String originalTrainingDataFile) |
IClassifier |
train(List<Tuple> trainingData) |
static void |
writeToFile(List<Tuple> dataset,
String filename) |
public NaiveBayesClassifier(InputStream modelInputStream)
public NaiveBayesClassifier()
public void setParameter(Properties props)
setParameter in interface IClassifierpublic Map<String,Double> predict(Tuple predict)
predict in interface IClassifierpublic IClassifier train(List<Tuple> trainingData)
train in interface IClassifierpublic void persistModel(String filename)
persistModel in interface IClassifierpublic void loadModel(InputStream inputStream)
loadModel in interface IClassifierpublic static void splitData(String originalTrainingDataFile)
Copyright © 2015. All rights reserved.