History в AutoKeras

Рейтинг: 0Ответов: 0Опубликовано: 15.03.2023

У меня есть объект класса autokerаs, который автоматически строит нейронку модели регрессии, обучается на подаваемом датасете и записывает историю обучения. Хотелось бы посмотерть историю обучения каждой попытки (в даном случае их три, т.к. max_trials=3). Но строчка hist = pd.DataFrame(REGG.history) сохраняет только полседнюю историю обучения. Подскажите, нет ли возможности это модифицировать, чтобы просматривать историю обучения любой попытки? Заранее спасибо за ответ.

import autokeras as ak
train_set = tf.data.Dataset.from_tensor_slices((X_train, Y_train))
test_set = tf.data.Dataset.from_tensor_slices((X_test, Y_test))

reg2 = ak.StructuredDataRegressor(max_trials=3,
                                  overwrite=True,
                                  project_name="AutoKeras4",
                                  loss='mean_absolute_error',
                                  metrics = 'accuracy'
                                  )
# Feed the tensorflow Dataset to the regressor.
REGG = reg2.fit(train_set, verbose=True, epochs=50)

hist = pd.DataFrame(REGG.history)
hist['epoch'] = REGG.epoch

Ответы

Ответов пока нет.