com.hankcs.hanlp.classification.classifiers

Class NaiveBayesClassifier

java.lang.Object

com.hankcs.hanlp.classification.classifiers.AbstractClassifier

com.hankcs.hanlp.classification.classifiers.NaiveBayesClassifier

All Implemented Interfaces:

IClassifier

public class NaiveBayesClassifier
extends AbstractClassifier

实现一个基于多项式贝叶斯模型的文本分类器

Constructor Summary

Constructors

Constructor and Description
NaiveBayesClassifier() 构造一个空白的贝叶斯分类器，通常准备用来进行训练
NaiveBayesClassifier(NaiveBayesModel naiveBayesModel) 由训练结果构造一个贝叶斯分类器，通常用来加载磁盘中的分类器

Method Summary

Modifier and Type	Method and Description
double[]	categorize(Document document) 预测分类
AbstractModel	getModel() 获取训练后的模型,可用于序列化保存或预测.
NaiveBayesModel	getNaiveBayesModel() 获取训练结果
Map<String,Double>	predict(String text) 预测分类
protected BaseFeatureData	selectFeatures(IDataSet dataSet) 统计特征并且执行特征选择，返回一个FeatureStats对象，用于计算模型中的概率
void	train(IDataSet dataSet) 训练模型

Methods inherited from class com.hankcs.hanlp.classification.classifiers.AbstractClassifier

classify, classify, enableProbability, label, predict, train, train, train

Methods inherited from class java.lang.Object

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

Constructor Detail

NaiveBayesClassifier

public NaiveBayesClassifier(NaiveBayesModel naiveBayesModel)

由训练结果构造一个贝叶斯分类器，通常用来加载磁盘中的分类器

Parameters:

naiveBayesModel -

NaiveBayesClassifier

public NaiveBayesClassifier()

构造一个空白的贝叶斯分类器，通常准备用来进行训练

Method Detail

getNaiveBayesModel

public NaiveBayesModel getNaiveBayesModel()

获取训练结果

Returns:

train

public void train(IDataSet dataSet)

Description copied from interface: IClassifier

训练模型

Parameters:

dataSet - 训练数据集

getModel

public AbstractModel getModel()

Description copied from interface: IClassifier

获取训练后的模型,可用于序列化保存或预测.

Returns:

模型,null表示未训练

predict

public Map<String,Double> predict(String text)
                           throws IllegalArgumentException,
                                  IllegalStateException

Description copied from interface: IClassifier

预测分类

Parameters:

text - 文本

Returns:

所有分类对应的分值(或概率, 需要enableProbability)

Throws:

IllegalArgumentException - 参数错误

IllegalStateException - 未训练模型

categorize

public double[] categorize(Document document)
                    throws IllegalArgumentException,
                           IllegalStateException

Description copied from interface: IClassifier

预测分类

Returns:

Throws:

IllegalArgumentException

IllegalStateException

selectFeatures

protected BaseFeatureData selectFeatures(IDataSet dataSet)

统计特征并且执行特征选择，返回一个FeatureStats对象，用于计算模型中的概率

Parameters:

dataSet -

Returns: