Ошибка MLPRegressor при использовании Solver SGD - PullRequest
1 голос
/ 13 февраля 2020

Я создаю MLP в python с помощью sklearn.neural_network MLPRegressor.

У меня есть поиск по сетке:

param_grid={'hidden_layer_sizes': [(100,100), (50,50,50), (100,)],
            ....
            'solver':['adam', 'sgd']}

grid=GridSearchCV(MLPRegressor, param_grid, cv)
grid.fit(x_train, y_train)
...

Что я считаю действительно странным: если я удаляю решатель в param_grid и выбран adam в качестве решателя, все работает отлично.

Однако я хочу использовать sgd в качестве решателя. Как только я использую это в параметре Grid и больше ничего не меняю, я получаю ошибку:

Ошибка значения: вход содержит NAN, бесконечность или значение, слишком большое для dtype ('float64') для line grid.fit

Я проверил свой ввод: нет значений Nan, бесконечности и нормальных значений в диапазоне от 0 до 1.

Почему это

...