2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Removed metrics that are unavailable in tensorflow 1.12

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-17 18:30:04 +02:00
parent 2b64209bda
commit 9a5c96c46c
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/twomartens/masterthesis/ssd.py
View File

@ -475,10 +475,7 @@ def train_keras(train_generator: callable,
beta1=0.5, beta2=0.999), beta1=0.5, beta2=0.999),
loss=ssd_loss.compute_loss, loss=ssd_loss.compute_loss,
metrics=[ metrics=[
tf.keras.metrics.Precision(), "categorical_accuracy"
tf.keras.metrics.Recall(),
tf.keras.metrics.FalsePositives(),
tf.keras.metrics.CategoricalAccuracy()
] ]
) )