| { | |
| "architectures": [ | |
| "SimplePyTorchTabularClassifier" | |
| ], | |
| "model_type": "tabular_classification", | |
| "framework": "pytorch", | |
| "input_dim": 10, | |
| "num_labels": 3, | |
| "id2label": { | |
| "0": "class_A", | |
| "1": "class_B", | |
| "2": "class_C" | |
| } | |
| } |
| { | |
| "architectures": [ | |
| "SimplePyTorchTabularClassifier" | |
| ], | |
| "model_type": "tabular_classification", | |
| "framework": "pytorch", | |
| "input_dim": 10, | |
| "num_labels": 3, | |
| "id2label": { | |
| "0": "class_A", | |
| "1": "class_B", | |
| "2": "class_C" | |
| } | |
| } |