Как получить весовые матрицы во время тренировок на Scikit - PullRequest
Новая
       3

Как получить весовые матрицы во время тренировок на Scikit

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

Я тренирую MLPClassifier с помощью Scikit. Допустим, я хочу тренироваться в течение 5 эпох на MNIST с одним скрытым слоем из 100 нейронов.

Если я сделаю «mlp = MLPClassifier (...)», а затем «mlp.fit (train, test)», тогда я могу получить обученные веса с помощью «mlp.coefs _».

Но мне нужна последовательность весовых матриц, полученных после каждой эпохи во время тренировки. Поэтому, если я тренируюсь в течение 5 эпох, мне нужен список размером 5 с историей весовых матриц.

Возможно ли это с помощью Scikit? Или я должен использовать Keras?

1 Ответ

1 голос
/ 25 января 2020

Один из вариантов - тренировать вашу модель с долей эпох, которые вы хотели сделать.

Сохраните параметры.

Затем продолжите тренировку вашей модели с параметром warm_start = True. Вы будете делать это до тех пор, пока не получите желаемое общее количество эпох.

В контексте реализации изучения sci-kit параметром max_iter будут эпохи. Это ссылка в этой ссылке.
https://stats.stackexchange.com/questions/284491/are-the-epochs-equivalent-to-the-iterations

...