可以直接用fit過的GridSearchCV來做predict嗎？ - Cupoy

2019/06/05 下午 10:37

機器學習共學討論版

Ken Hu

觀看數：3

回答數：1

收藏數：0

ml100-2

在做homework 47的時候

一開始先用GridSearchCV做fit

然後根據所找到的最佳參數，重新建立分類器

再對新的分類器做訓練、做預測

但GridSearchCV本身也有predict()這個method

它的說明寫：「Call predict on the estimator with the best found parameters」

但我試著用：

grid_search = GridSearchCV( ... )  

search_result = grid_search.fit(X_train, y_train)

y_pred = search_result.predict(X_test)

print(metrics.accuracy_score(y_test, y_pred))

輸出的結果郤是一開始用預設值建立分類器的結果，而不是使用了最佳參數的結果

那麼，到底GridSearchCV本身的predict()這個method的用意是什麼呢？

回答列表

• 2019/06/06 下午 00:05
  張維元 (WeiYuan)
  贊同數：1
  不贊同數：0
  留言數：0

  嗨，直接使用 search_result.predict 應該就是利用其找出的最佳參數下去預測的結果。這邊可能看一下你的 GridSearchCV 中 estimator 怎麼設置？

  
  

  補充一下，GridSearchCV 會針對 estimator 沒有指定的參數做最佳化的搜尋。