Skip to content

Comments

ML Pipeline v2#67

Merged
mihow merged 22 commits intomainfrom
feat/return-all-scores
Jan 26, 2025
Merged

ML Pipeline v2#67
mihow merged 22 commits intomainfrom
feat/return-all-scores

Conversation

@mihow
Copy link
Collaborator

@mihow mihow commented Nov 23, 2024

  • Update API schema to match latest features to support the new self-service ML Pipeline registry in Antenna
  • New /info endpoint that returns available Algorithms and Pipelines, and the server status
  • Support for serving Category Maps, aka labels or classes for the classification models (also provided via the /info endpoint and with each batch of classification results, but not with every single detection)
  • Return all raw logits from the model prediction with each detection classification response
  • Return all softmax scores from the model prediction with each detection classification response
  • Some updates to Pydantic syntax for v2
  • Add tests for the API and workflow for running them in Github

This will be deployed with the corresponding Antenna PR in RolnickLab/antenna#684

@mihow mihow self-assigned this Jan 16, 2025
@mihow mihow changed the title Option to return scores for all classes in a prediction ML Pipeline v2 Jan 26, 2025
@mihow mihow merged commit 8e2c885 into main Jan 26, 2025
3 checks passed
@mihow mihow mentioned this pull request Jan 26, 2025
5 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant