From 9a5c96c46c9dcaa0aa8373c992c721b58ece38ed Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 17 Jun 2019 18:30:04 +0200 Subject: [PATCH] Removed metrics that are unavailable in tensorflow 1.12 Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 81ed4c4..9cf96af 100644 --- 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" ] )