com.hankcs.hanlp.classification.classifiers

类 NaiveBayesClassifier

java.lang.Object

com.hankcs.hanlp.classification.classifiers.AbstractClassifier

com.hankcs.hanlp.classification.classifiers.NaiveBayesClassifier

所有已实现的接口:

IClassifier

public class NaiveBayesClassifier
extends AbstractClassifier

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

构造器概要

构造器

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

方法概要

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

从类继承的方法 com.hankcs.hanlp.classification.classifiers.AbstractClassifier

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

从类继承的方法 java.lang.Object

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

构造器详细资料

NaiveBayesClassifier

public NaiveBayesClassifier(NaiveBayesModel naiveBayesModel)

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

参数:

naiveBayesModel -

NaiveBayesClassifier

public NaiveBayesClassifier()

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

方法详细资料

getNaiveBayesModel

public NaiveBayesModel getNaiveBayesModel()

获取训练结果

返回:

train

public void train(IDataSet dataSet)

从接口复制的说明: IClassifier

训练模型

参数:

dataSet - 训练数据集

getModel

public AbstractModel getModel()

从接口复制的说明: IClassifier

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

返回:

模型,null表示未训练

predict

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

从接口复制的说明: IClassifier

预测分类

参数:

text - 文本

返回:

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

抛出:

IllegalArgumentException - 参数错误

IllegalStateException - 未训练模型

categorize

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

从接口复制的说明: IClassifier

预测分类

返回:

抛出:

IllegalArgumentException

IllegalStateException

selectFeatures

protected BaseFeatureData selectFeatures(IDataSet dataSet)

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

参数:

dataSet -

返回: