您好,登錄后才能下訂單哦!
在Python中,有幾個常用的庫和函數可用于評估機器學習模型。以下是一些建議:
accuracy_score
來計算準確率,confusion_matrix
來生成混淆矩陣,classification_report
來生成分類報告等。plot_confusion_matrix
函數來繪制混淆矩陣,plot_roc_curve
函數來繪制ROC曲線等。DataFrame.describe()
方法來獲取數據的描述性統計信息,DataFrame.groupby()
方法來進行分組分析等。f1_score
、precision_score
、recall_score
等。這些函數可以幫助你更全面地評估模型的性能。cross_val_score
,網格搜索函數GridSearchCV
等。以下是一個簡單的例子,展示了如何使用scikit-learn庫來評估一個分類模型:
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
import matplotlib.pyplot as plt
import seaborn as sns
# 加載數據集
iris = load_iris()
X = iris.data
y = iris.target
# 劃分訓練集和測試集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 訓練模型
clf = RandomForestClassifier(n_estimators=100, random_state=42)
clf.fit(X_train, y_train)
# 預測
y_pred = clf.predict(X_test)
# 評估模型
print("Accuracy:", accuracy_score(y_test, y_pred))
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred))
print("Classification Report:\n", classification_report(y_test, y_pred))
# 繪制混淆矩陣
sns.heatmap(confusion_matrix(y_test, y_pred), annot=True, fmt='d')
plt.xlabel('Predicted')
plt.ylabel('True')
plt.show()
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。