Средние веса из двух папок .h5 в кератах - PullRequest
0 голосов
/ 08 февраля 2020

Я обучил две модели для разных наборов данных и сохранил веса каждой модели как ModelA.h5 и ModelB.h5. Я хочу усреднить эти веса и создать новую папку с именем Model C .h5 и загрузить ее в одну модель. Architechture. Как мне это сделать?

1 Ответ

0 голосов
/ 08 февраля 2020

Модель, обученную на разных наборах данных, просто не может быть добавлена ​​вот так. Это выглядит примерно так. Скажем так: обучите одного человека классифицировать 1000 изображений на 5 классов, затем обучите другого человека классифицировать еще 1000 изображений на те же 5 классов. Теперь вы хотите объединить их в одну.

Скорее всего, вы можете взять ансамбль обеих сетей. Существует несколько способов объединить прогнозы обеих моделей, используя Max Voting, Averaging или Weighted Average, Bagging and Boosting и т. Д. c. Ансамбль помогает превратить слабые классификаторы в один сильный классификатор.

Вы можете обратиться к этой ссылке, чтобы узнать больше о различных типах ансамбля: Link

...