Scikit-Learn 작업의 일부로 keras.wrappers.scikit_learn.py
에서 찾을 수 있는 wrappers를 통해 Sequential
Keras 모델(하나의 입력만)을 사용할 수 있습니다.
사용 가능한 wrappers는 두 가지가 있습니다:
keras.wrappers.scikit_learn.KerasClassifier(build_fn=None, **sk_params)
는 Scikit-Learn 분류기 인터페이스,
keras.wrappers.scikit_learn.KerasRegressor(build_fn=None, **sk_params)
는 Scikit-Learn 회귀 인터페이스 입니다.
인수
build_fn
는 Keras 모델을 생성하고 컴파일하고 반환하여 피팅이나 예측에 사용할 수 있도록 합니다. 다음 세 가지 값 중 하나가 build_fn으로 넘겨질 수 있습니다:
KerasClassifier
나 KerasRegressor
를 물려받은 클래스를 구현한다는 의미입니다. 현재 클래스의 call 메소드는 build_fn의 기본값으로 취급됩니다.sk_params
는 모델 파라미터와 피팅 파라미터 양쪽을 다 받습니다. 허용된 모델 파라미터는 build_fn
의 인수입니다. scikit-learn의 다른 모든 예측기와 같이 'build_fn'은 자신의 인수의 기본값을 제공하여 sk_params
에 어떤 값도 넘겨주지 않고도 예측기를 만들 수 있습니다.
sk_params
는 fit
, predict
,predict_proba
, score
메소드를 호출하여 파라미터를 받을 수도 있습니다(예를 들어, epochs
, batch_size
). 피팅(예측) 마라미터는 다음과 같은 순서로 선택됩니다:
fit
, predict
, predict_proba
, score
메소드의 인덱스 집합 인수로 넘겨진 값.sk_params
로 넘겨진 값.keras.models.Sequential
의 fit
, predict
, predict_proba
, score
메소드의 기본값.Scikit-learn의 grid_search
API를 사용할 때, 허용된 조정 가능한 파라미터는 피팅 파라미터를 포함해서sk_params
로 넘겨줄 수 있는 파라미터입니다. 즉, grid_search
를 모델 파라미터를 찾기 위해서 뿐만 아니라 최적의 batch_size
나 epochs
를 찾기 위해서 사용할 수 있습니다.
이 문서는 Keras의 Scikit-learn API을 번역한 것입니다.