如何以不同的方式打印模型的准确性



我正在进行监督式机器学习。目前,通过使用scikit的指标,它可以打印出整个语料库的准确性。

我还希望打印出前 3 个主题

的准确性,然后打印出前 5 个主题的准确性。我该怎么做?

model = LogisticRegression()
model = model.fit(matrix, label)
y_train_pred = model1.predict(matrix_test)
print(metrics.accuracy_score(label_test, y_train_pred))

您可以使用混淆矩阵: http://scikit-learn.org/stable/modules/generated/sklearn.metrics.confusion_matrix.html

示例:http://scikit-learn.org/stable/auto_examples/model_selection/plot_confusion_matrix.html

通过这种方式,您可以获得应用于每个类别预测的特定信息。

最新更新