我正在嘗試從我的GridSearchCV物件中列印最佳模型的訓練和測驗分數。我最初的猜測是使用cv_results['best_train_score']，cv_results['best_test_score'] 但在查看檔案后，我認為 cv_results 沒有“best_train_score”。
我也看到有一個，best_estimator_但我不確定我是否可以用它來列印測驗和訓練分數。任何幫助是極大的贊賞。
uj5u.com熱心網友回復：
您可以使用best_estimator_擬合的GridSearchCV來檢索最佳模型，然后使用score估計器的功能來計算訓練和測驗精度。
如下：
from sklearn import svm, datasets
from sklearn.model_selection import GridSearchCV, train_test_split
iris = datasets.load_iris()
X_train, X_test, y_train, y_test = train_test_split(
iris.data, iris.target, test_size=0.2
)
parameters = {"kernel": ("linear", "rbf"), "C": [1, 10]}
svc = svm.SVC()
cv = GridSearchCV(svc, parameters)
cv.fit(iris.data, iris.target)
model = cv.best_estimator_
print(f"train score: {model.score(X_train, y_train)}")
print(f"test score: {model.score(X_test, y_test)}")
輸出：
train score: 0.9916666666666667
test score: 1.0
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/324924.html