决策树模型召回率计算方法 - sklearn.metrics.recall_score
要查看决策树模型 clf 的召回率,可以使用 sklearn.metrics 模块中的 recall_score 函数。这个函数需要传入两个参数:真实标签和预测标签。可以使用 clf.predict 函数来得到预测标签,使用测试数据集的真实标签来计算召回率。
示例代码:
from sklearn.metrics import recall_score
# X_test 是测试数据集特征矩阵,y_test 是测试数据集的真实标签
y_pred = clf.predict(X_test)
recall = recall_score(y_test, y_pred)
print('Recall: ', recall)
这个代码会输出决策树模型 clf 的召回率。
原文地址: https://www.cveoy.top/t/topic/oqci 著作权归作者所有。请勿转载和采集!