search

Tencent / NeuralNLP-NeuralClassifier

An Open-source Neural Hierarchical Multi-label Text Classification Toolkit
Other
1.85k stars 406 forks source link

能否输出不同类别的概率 #85

Closed wingsyuan closed 3 years ago

wingsyuan commented 3 years ago

请问， 这里面的所有的分类模块都能输出预测类别的概率吗？我的应用场景需要预测类别的概率值。。

coderbyr commented 3 years ago

可以，predict.py提供了对已训练完成的模型进行加载并进行预测的功能，predict函数输出的记为类别的预测概率，可以根据需要进行修改；