К какому классу относятся записи журнала в параметре on_epoch_end keras.callbacks.Callback - PullRequest
Новая
       20

К какому классу относятся записи журнала в параметре on_epoch_end keras.callbacks.Callback

1 голос
/ 05 июня 2019

Я пытаюсь получить оценки от моей модели Keras после каждой эпохи, поэтому я использую следующий пользовательский обратный вызов: 

class EpochScoreHistory(keras.callbacks.Callback):
    def __init__(self, keys):
        self.custom_keys = keys

    def on_train_begin(self, logs={}):
        self.custom_avg_scores = dict()
        for key in self.custom_keys:
            self.custom_avg_scores[key]=[]

    def on_epoch_end(self, epoch, logs={}):
        for key in self.custom_keys:
            self.custom_avg_scores[key].append(logs.get(key))

После каждой эпохи, внутри параметра logs функции on_epoch_end, у меня есть такие оценки, как acc, например, 0.567 и т. Д., Но мне интересно, к какому классу относятся эти оценки, так как в моей задаче есть два класса для классификации.

Короче говоря, когда мы получаем журналы в качестве параметра функции on_epoch_end настраиваемого обратного вызова Keras, к какому классу классификации относятся оценки в этих журналах? Например, я ожидал получить список точных значений, но вместо этого у меня есть отдельные значения для каждого результата, который я получаю. Интересно, это ошибка в Керасе? Большое спасибо.

1 Ответ

0 голосов
/ 05 июня 2019

Keras дает вам точность модели, т. Е. Насколько она хороша при классификации классов.Это не дает вам ценность для каждого класса.Возможно, вам придется создать матрицу путаницы для этого.

...