Files
Kordant/ml/spam-classifier/output/onnx_model/model_metadata.json

17 lines
299 B
JSON

{
"version": "1.0.0",
"model_name": "distilbert-base-uncased",
"task": "sms-spam-classification",
"max_length": 128,
"num_labels": 2,
"label2id": {
"ham": 0,
"spam": 1
},
"id2label": {
"0": "ham",
"1": "spam"
},
"framework": "pytorch",
"export_format": "onnx"
}