Skip to content

Commit 7a615a5

Browse files
committed
fix: complete rename of softmax_scores field (now any calibrated score)
1 parent 4f81651 commit 7a615a5

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
@@ -339,7 +339,7 @@ def save_results(results: PipelineResponse | None = None, results_json: str | No
339339
defaults={
340340
"timestamp": classification.timestamp or now(),
341341
"logits": classification.logits,
342-
"softmax_scores": classification.scores,
342+
"scores": classification.scores,
343343
"category_map": classification_algo.category_map,
344344
},
345345
)

0 commit comments

Comments
 (0)