Используйте обученный вес для обучения другого набора данных

Question

Я обучил модель CNN на наборе данных, который содержит 4 выходных класса, я сохранил веса модели в "weights.h5". Я хочу использовать эти веса для обучения другого набора данных, который содержит только 2 выходных класса, как я могу это сделать?

Answer 1

1. Загрузить модель / вес
2. Удалить последний слой из обученной сети
3. Присоединить новый выходной слой
4. Рекомпилированные

base_model.load_weights(...)

x = base_model.layers[-2].output
x = Dense(2,activation='softmax')(x) # 2 output classes
model = Model(base_model.input,x)

model.compile(...)