Я использую генератор данных k-folder вручную для обучения моей нейронной сети, созданной Keras. Ниже описан процесс:
def build_model():
encoder = Dense(...)(input)
encoder = Dense(...)(encoder)
decoder = Dense(...)(encoder)
decoder = Dense(...)(decoder)
autoencoder = Model(input_layer, decoder)
autoencoder.compile(...)
return autoencoder
def train_model(...):
autoencoder = build_model()
autoencoder.fit()
return autencoder
for train_index, test_index in KFold(X, 10):
trained_model = train_model(X_train, X_test)
Это простой процесс, который я использовал. Моя путаница в том, что когда я делаю этот тренировочный процесс в l oop, я чувствую, что нейронная сеть не инициализируется каждый раз. Похоже, что тренировочный процесс продолжается на основе предыдущего результата, обученные веса каждого нейрона все еще находятся в памяти.
Мне интересно, есть ли способ очистить память для каждой тренировки? oop? Или этот процесс должен быть в порядке, и я могу тренировать эту модель снова и снова?