Как экспортировать настроенные метрики из CustomCallBack в Tensorflow2 - PullRequest
Новая
       5

Как экспортировать настроенные метрики из CustomCallBack в Tensorflow2

0 голосов
/ 12 июля 2020

У меня есть модель, обучающая мультикатегориям.

Я написал CustomCallback, который печатает метрики для каждого epoch_end 

custom_metrics = []
class CustomCallback(tf.keras.callbacks.Callback):
    def on_epoch_end(self, epoch, logs={}):
        preds = self.model.predict(self.valid_inputs)
        preds0 = np.argmax(preds,axis=1)
        num_outputs = np.argmax(self.valid_outputs,axis=1)
        print("\n")
        precision, recall, fscore, support = precision_recall_fscore_support(num_outputs, preds0, average=None)
        print_metrics([precision, recall, fscore, support])
        custom_metrics.append([precision, recall, fscore, support])

Он печатает нормально, но я хотел бы получить [точность, Напомним, fscore, support] в глобальной области видимости, чтобы иметь возможность изобразить эволюцию после обучения. конец обучения.

Как я мог это сделать?

...