在Scikit-learn中,可以使用metrics
模塊中的各種函數(shù)來實(shí)現(xiàn)模型評估指標(biāo)。常用的模型評估指標(biāo)包括準(zhǔn)確率(accuracy)、精確率(precision)、召回率(recall)、F1值、ROC曲線和AUC等。
以下是一些常用的模型評估指標(biāo)函數(shù):
accuracy_score(y_true, y_pred)
precision_score(y_true, y_pred)
recall_score(y_true, y_pred)
f1_score(y_true, y_pred)
roc_curve(y_true, y_score)
, roc_auc_score(y_true, y_score)
其中,y_true
為真實(shí)標(biāo)簽,y_pred
為預(yù)測標(biāo)簽,y_score
為決策函數(shù)或概率預(yù)測值。
示例代碼:
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, roc_curve, roc_auc_score
# 計算準(zhǔn)確率
accuracy = accuracy_score(y_true, y_pred)
# 計算精確率
precision = precision_score(y_true, y_pred)
# 計算召回率
recall = recall_score(y_true, y_pred)
# 計算F1值
f1 = f1_score(y_true, y_pred)
# 計算ROC曲線和AUC值
fpr, tpr, thresholds = roc_curve(y_true, y_score)
auc = roc_auc_score(y_true, y_score)
通過這些函數(shù),可以方便地對模型進(jìn)行評估,并選擇最合適的評估指標(biāo)來評價模型的性能。