fix(ml-service): test logic + prediction_response status forward

- calibrate_probs_selects_top_label: use logits that produce confidence >= 0.70
- prediction_response: forward status from Prediction struct instead of hardcode

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
MythEclipse
2026-06-12 17:08:27 +00:00
co-authored by Claude
parent 9ab472ce1b
commit c116a2c4a5
2 changed files with 5 additions and 3 deletions
+1 -1
View File
@@ -61,7 +61,7 @@ pub fn metadata_response(model_path: String, model_loaded: bool, input_size: u32
pub fn prediction_response(prediction: Prediction) -> PredictionResponse {
PredictionResponse {
status: "ok".to_string(),
status: prediction.status,
label: prediction.label,
confidence: prediction.confidence,
probabilities: prediction.probabilities,