Метрики «Точность» и «Отзыв» для каждого класса в задаче сегментации. - PullRequest
0 голосов
/ 16 июня 2020

Я обучаю модель сегментации (более 1d сигнала), в которой у меня есть два разных возможных класса для каждого «пикселя» или «координаты».

Я использую двоичную разреженную перекрестную потерю энтропии. Мой последний слой - это слой softmax с размерами (N, 2), когда N - количество «пикселей» в моих данных.

self.out = Dense(2 * features_num, activation='relu') 

Мне интересно наблюдать за «Точностью» и «Отзыв» статистика для каждого из классов (1 и 0) после каждой эпохи. Однако я не мог найти более простого способа сделать это, чем написать это сам.

Кто-нибудь знает простое решение для этого?

Я использую TensorFlow 2.x и использую API высокого уровня для обучения (model.fit).

Спасибо за лот, Эяль.

...