from sklearn.metrics import r2_score
r2_score(y,pred_y)
最后举个例子,或者说方便大家copy,
比如对于GaussNaiveBayes,经典的分类模型,我们可以:
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score
print('Accuracy score: ', accuracy_score(y_test, predictions))
print('Precision score: ', precision_score(y_test, predictions))
print('Recall score: ', recall_score(y_test, predictions))
print('F1 score: ', f1_score(y_test, predictions))
>>> Accuracy score: 0.9885139985642498
>>> Precision score: 0.9720670391061452
>>> Recall score: 0.9405405405405406
>>> F1 score: 0.9560439560439562