前提・実現したいこと
機械学習モデルの一つであるSVCのパラメータ設定を最適化するためにGridSearchCVを使用していますが
計算が終了しません。。。
パラメーター設定ミス等ございましたらご指摘お願いいたします。
該当のソースコード
python
1from sklearn.model_selection import GridSearchCV 2from sklearn.preprocessing import StandardScaler 3from sklearn.decomposition import PCA 4from sklearn.svm import SVC 5from sklearn.pipeline import make_pipeline 6 7''' 8(省略) 9''' 10 11pipe_svc = make_pipeline(StandardScaler(), 12 PCA(n_components=3), 13 SVC(random_state=1)) 14 15 16print('X_train.shape={}, y_train.shape={}'.format(X_train.shape, y_train.shape)) 17#X_train.shape=(74, 27), y_train.shape=(74,) ---------------訓練データ構造を出力 18 19param_grid = [{'svc__C': [0.1, 1.0, 10.0], 20 'svc__kernel': ['rbf']}] 21gs = GridSearchCV(estimator=pipe_svc, 22 param_grid=param_grid, 23 cv=2, 24 n_jobs=-1) 25 26gs = gs.fit(X_train, y_train) #----この部分で計算が終わらない 27 28print(gs.best_score_) 29print(gs.best_params_)
試したこと
下記のパラメーターの数を減らしましたが効果が出ませんでした。
make_pipeline中の PCA(n_components=3)
param_grid中の 'svc__C': [0.1, 1.0, 10.0]
GridSearchCV中の cv=2
補足情報(FW/ツールのバージョンなど)
google colabで計算
コードは下記URLを参考にしました
https://github.com/rasbt/python-machine-learning-book-3rd-edition/blob/master/ch06/ch06.ipynb
(Fine-tuning machine learning models via grid searchの章)
あなたの回答
tips
プレビュー