Removed metrics that are unavailable in tensorflow 1.12

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-17 18:30:04 +02:00
parent 2b64209bda
commit 9a5c96c46c
1 changed files with 1 additions and 4 deletions
--- a/src/twomartens/masterthesis/ssd.py
+++ b/src/twomartens/masterthesis/ssd.py
@ -475,10 +475,7 @@ def train_keras(train_generator: callable,
                                         beta1=0.5, beta2=0.999),
        loss=ssd_loss.compute_loss,
        metrics=[
-            tf.keras.metrics.Precision(),
-            tf.keras.metrics.Recall(),
-            tf.keras.metrics.FalsePositives(),
-            tf.keras.metrics.CategoricalAccuracy()
+            "categorical_accuracy"
        ]
    )