Skip to content

Commit

Permalink
adjust
Browse files Browse the repository at this point in the history
  • Loading branch information
gferraro committed Oct 1, 2024
1 parent 5ccea12 commit 166c252
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions src/modelevaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,12 +238,7 @@ def metadata_confusion(dir, confusion_file, after_date=None, model_metadata=None
model_meta = json.load(t)
labels = model_meta.get("labels", [])
excluded_labels = model_meta.get("excluded_labels", {})
remapped = model_meta.get("remapped_labels", {})
remapped_labels = {}
# slightly different format than from thermaldataset
for mapped_to, mapped_labels in remapped.items():
for mapped_label in mapped_labels:
remapped_labels[mapped_label] = mapped_to
remapped_labels = model_meta.get("remapped_labels", {})
else:
labels = [
"bird",
Expand Down

0 comments on commit 166c252

Please sign in to comment.