개발/머신러닝

파이썬.머신러닝 - Grid Search

웅'jk 2022. 12. 2. 15:46

그리드 서치란 ? 인공지능을 학습을 시킬때 어떤 학습법이 더 좋은지 찾아내는 방법이다.

 

param_grid = {'kernel':['linear','rbf','poly'] , 'C' : [0.1,1,10],
             'gamma':[0.01,0.1,1]
             }
# kernel은 모델, c 와 gamma 는 조절 수치이다.

from sklearn.model_selection import GridSearchCV
grid = GridSearchCV(SVC() , param_grid , refit=True, verbose=4)
grid.fit(X_train,y_train)

#최고의 인공지능
grid.best_estimator_
#최고의 값
grid.best_params_
#학습중최고적중률
grid.best_score_