Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

removed obsolete code

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-01 21:13:15 +02:00
parent 94783c54f2
commit 070749880e
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
image_prediction/redai_adapter/model_wrapper.py
View File

@ -66,10 +66,6 @@ class EfficientNetWrapper:
model = tf.keras.models.Model(inputs=input_img, outputs=finetuned) model = tf.keras.models.Model(inputs=input_img, outputs=finetuned)
model.compile( model.compile()
loss="categorical_crossentropy",
optimizer="adam",
metrics=[tf.keras.metrics.Recall(), tf.keras.metrics.Precision()],
)
return model return model