組み込み開発のためのPython · レッスン 140 / 220
グリッドサーチで設定を比較する
グリッドサーチは指定したハイパーパラメーターの組合せを交差検証で評価します。探索範囲を広げすぎず、最終テストデータを探索に使わないでください。
グリッドサーチは指定したハイパーパラメーターの組合せを交差検証で評価します。探索範囲を広げすぎず、最終テストデータを探索に使わないでください。 仕組みだけでなく、検証方法と運用上の注意も具体例で確認します。
基本の考え方
グリッドサーチは指定したハイパーパラメーターの組合せを交差検証で評価します。探索範囲を広げすぎず、最終テストデータを探索に使わないでください。
安全に試すための確認
例は学習用データを使うもので、品質を保証する予測器ではありません。実データの欠損、単位、偏り、データ漏えいを確認し、テスト用データを学習中に参照しないでください。
コード例
手元のPython環境に貼り付ける前に、使用しているPythonの版と対象ライブラリの対応を確認してください。
from sklearn.model_selection import GridSearchCV
from sklearn.tree import DecisionTreeClassifier
search = GridSearchCV(DecisionTreeClassifier(random_state=0), {"max_depth": [1, 2, 3]}, cv=2)
search.fit([[0], [1], [2], [3], [4], [5]], [0, 0, 0, 1, 1, 1])
print(search.best_params_)手を動かす課題
最大深さの候補を変えて探索し、交差検証スコアのばらつきも確認してください。