discriminator accuracy

1 files changed, 11 insertions, 0 deletions

diff --git a/become_yukarin/updater.py b/become_yukarin/updater.py
index 02ea5d3..f6444d0 100644
--- a/become_yukarin/updater.py
+++ b/become_yukarin/updater.py
@@ -69,6 +69,17 @@ class Updater(chainer.training.StandardUpdater):
             reporter.report({'fake': loss_dis_f}, self.discriminator)
             reporter.report({'true': loss_dis_t}, self.discriminator)
 
+            tp = (d_true.data > 0.5).sum()
+            fp = (d_fake.data > 0.5).sum()
+            fn = (d_true.data <= 0.5).sum()
+            tn = (d_fake.data <= 0.5).sum()
+            accuracy = (tp + tn) / (tp + fp + fn + tn)
+            precision = tp / (tp + fp)
+            recall = tp / (tp + fn)
+            reporter.report({'accuracy': accuracy}, self.discriminator)
+            reporter.report({'precision': precision}, self.discriminator)
+            reporter.report({'recall': recall}, self.discriminator)
+
         loss = {'predictor': loss_l1 * self.loss_config.l1}
 
         if self.aligner is not None: