Skip to content

Commit 9f10aa6

Browse files
committed
fix: complete rename of softmax_scores field (now any calibrated score)
1 parent f85dd54 commit 9f10aa6

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

ami/main/api/serializers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -719,7 +719,7 @@ class Meta:
719719
"taxon",
720720
"score",
721721
"algorithm",
722-
"softmax_scores",
722+
"scores",
723723
"logits",
724724
"created_at",
725725
]

ami/ml/models/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def save_results(results: PipelineResponse | None = None, results_json: str | No
340340
defaults={
341341
"timestamp": classification.timestamp or now(),
342342
"logits": classification.logits,
343-
"softmax_scores": classification.scores,
343+
"scores": classification.scores,
344344
"category_map": classification_algo.category_map,
345345
},
346346
)

0 commit comments

Comments
 (0)