pytorch-tabular-legacy / config.json
Khauneesh's picture
Upload folder using huggingface_hub
47db207 verified
{
"architectures": [
"SimplePyTorchTabularClassifier"
],
"model_type": "tabular_classification",
"framework": "pytorch",
"input_dim": 10,
"num_labels": 3,
"id2label": {
"0": "class_A",
"1": "class_B",
"2": "class_C"
}
}