Create a proof_prediction
table to store predictions from ML models
#511
Labels
proof_prediction
table to store predictions from ML models
#511
We now have a new ML model that can be used to predict automatically the kind of proof image the user sends us.
For every newly uploaded proof, we want to call this endpoint to classify the proof type and store it in the DB.
For this, I suggest that we create a new
proof_prediction
table that has the following fields:id
(auto-incrementing int)proof_id
: the ID of the proof associated with the predictiontype
: str, the task type (object_detection
orclassification
, hereclassification
)model_name
: str, name of the ML model, hereprice_proof_classification
model_version
: str, version of the model, hereprice_proof_classification_1.0
created
: datetime of creation of the predictiondata
: JSONB, dictionary containing prediction data (here what's returned by the API)value
: str, the predicted value, for classification only (otherwise null)max_confidence
: float, the maximum score of the prediction, for classification it's the score of the predicted class, for object detection it's the maximum confidence score for all objects.We don't have (yet) object detection models, but it will come soon :)
This is just a proposal, feel free to comment!
poke @raphodn
The text was updated successfully, but these errors were encountered: