EarlyStopping в GridSearch - как получить среднюю эпоху, после которой обучение прекратилось? - PullRequest
Новая
       59

EarlyStopping в GridSearch - как получить среднюю эпоху, после которой обучение прекратилось?

0 голосов
/ 30 января 2020

есть ли способ получить среднее количество эпох, когда обучение прекращено с помощью EarlyStopping в GridSearch? 

grid = GridSearch(estimator=model, param_grid=params, cv=10, scoring='accuracy', refit='acc')
results = grid.fit(X, y, epochs=100, valiation_data=(X_val, y_val), callbacks=[EarlyStopping(monitor='val_loss', patience=10, verbose=0, mode='min', restore_best_weights=True)])

Примерно так: 

for mean, stdev, param in zip(grid.cv_results_['mean_test_acc'],
                              grid.cv_results_['std_test_acc'],
                              grid.cv_results_['params'],
                              grid.cv_results_['mean_epochs' ?????

Большое спасибо за каждый намек. Пожалуйста, не подведи меня

...