learning_rate is not a legal parameter

핸즈온 머신러닝(handsonml) 소스 실행 중 위와 같은 에러가 발생해서 해결했다.

 

from scipy.stats import reciprocal
from sklearn.model_selection import RandomizedSearchCV

param_distribs = {
    "n_hidden": [0, 1, 2, 3],
    "n_neurons": np.arange(1, 100)               .tolist(),
    "lr": reciprocal(3e-4, 3e-2)      .rvs(1000).tolist(),
}

rnd_search_cv = RandomizedSearchCV(keras_reg, param_distribs, n_iter=10, cv=3, verbose=2)
rnd_search_cv.fit(X_train, y_train, epochs=100,
                  validation_data=(X_valid, y_valid),
                  callbacks=[keras.callbacks.EarlyStopping(patience=10)])

RandomizedSearchCV에 들어가는 param_distribs의 "learning_rate" > "lr"로 수정하니까 정상 실행됐다.

실행환경 : tensorflow 2.3.0