Почему RandomizedSearchCV выдает ошибку, когда я использую ее для поиска параметров в TF 2.0? - PullRequest
Новая
       4

Почему RandomizedSearchCV выдает ошибку, когда я использую ее для поиска параметров в TF 2.0?

0 голосов
/ 03 марта 2020

Мой код как таковой, я пытаюсь найти параметры в TF 2.0, используя API KerasRegressor. 

def build_model(hidden_layers = 1,
                layer_size = 30,
                learning_rate = 3e-3):
    model = keras.models.Sequential()
    model.add(keras.layers.Dense(layer_size, activation='relu',
                                 input_shape=x_train.shape[1:]))
    for _ in range(hidden_layers - 1):
        model.add(keras.layers.Dense(layer_size,
                                     activation = 'relu'))
    model.add(keras.layers.Dense(1))
    optimizer = keras.optimizers.SGD(learning_rate)
    model.compile(loss = 'mse', optimizer = optimizer)
    return model

sklearn_model = tf.keras.wrappers.scikit_learn.KerasRegressor(
    build_fn = build_model)
callbacks = [keras.callbacks.EarlyStopping(patience=5, min_delta=1e-2)]
history = sklearn_model.fit(x_train_scaled, y_train,
                            epochs = 10,
                            validation_data = (x_valid_scaled, y_valid),
                            callbacks = callbacks)

from scipy.stats import reciprocal
param_distribution = {
    "hidden_layers":[1, 2, 3, 4],
    "layer_size": np.arange(1, 100),
    "learning_rate": reciprocal(1e-4, 1e-2),
}

from sklearn.model_selection import RandomizedSearchCV

random_search_cv = RandomizedSearchCV(sklearn_model,
                                      param_distribution,
                                      n_iter = 10,
                                      cv = 3,
                                      n_jobs = 1)
random_search_cv.fit(x_train_scaled, y_train, epochs = 100,
                     validation_data = (x_valid_scaled, y_valid),
                     callbacks = callbacks)

Однако в результате возникает ошибка:

Невозможно клонировать объект, так как конструктор не устанавливает или не изменяет параметр layer_size

На мой взгляд, я содержал параметр layer_size.

Что я должен сделать, чтобы решить эту проблему? также, когда я изменяю "n_jobs"> 1, это не может работать.

...