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

时间:2016-01-24 03:12:11

标签: python scikit-learn

我正在进行有监督的机器学习。目前,通过使用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))

1 个答案:

答案 0 :(得分:2)

您可以使用混淆矩阵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

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