评估措施神经网络模型性能

Question

我想创建一个将输入分为四类的神经网络模型：A，B，C和D类。每个类可以是0和1。我应该使用什么度量来评估我的模型。我可以使用Precision，Recall，F量度评估模型吗？

Answer 1

这取决于您拥有的数据，

1. 如果您的数据非常均衡（即所有类别的表示均等），则可以使用“准确性”作为度量。
2. 但是，如果您的数据正遭受班级不平衡（班级分布偏斜）的困扰，那么您应该考虑班级的精确度，回忆和F1得分。我建议f1得分，因为它同时考虑了准确性和召回率。
3. 否则，您可以绘制ROC曲线并获得ROC下的面积。

Sklearn有用于执行此操作的软件包，您可以参考classification report，Area under ROC 。