File size: 423 Bytes
3c3e10b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | {
"model_type": "kumo_tabular",
"default_size": "large",
"checkpoint_files": {
"small": {
"classification": "small/classifier.pt",
"regression": "small/regressor.pt"
},
"medium": {
"classification": "medium/classifier.pt",
"regression": "medium/regressor.pt"
},
"large": {
"classification": "large/classifier.pt",
"regression": "large/regressor.pt"
}
}
}
|