评估每个类的精度并在FastText中调用

时间:2017-11-28 12:53:37

标签: machine-learning precision-recall fasttext

我在this教程之后使用Facebook Research FastText库进行文本分类。我有2个标签,我正在进行分类(2级)。测试文件上的预测输出显示了相同的精度和召回率。如何计算每个类的精度并调用我的测试文件?

1 个答案:

答案 0 :(得分:1)

我最近自己不得不处理这件事。此issue in Github描述了该问题,并提供了solution

总之,您需要将此作为后处理步骤。上面链接的代码对您的实际标签与预测标签进行比较,并计算混淆矩阵,该矩阵准确地反映了分类器对二进制分类的性能。此代码仅计算混淆矩阵和精度。如果您还想添加精度和召回,您可以类似地使用scikitlearn API,例如sklearn.metrics.precision_recall_fscore_support