autism-detector / config.json
toderian's picture
Add config.json
11a9cc2 verified
{
"model_type": "tabular-binary-classifier",
"architecture": "feedforward",
"input_features": 8,
"hidden_sizes": [64, 32],
"num_labels": 2,
"id2label": {
"0": "Healthy",
"1": "ASD"
},
"label2id": {
"Healthy": 0,
"ASD": 1
},
"problem_type": "binary_classification",
"torch_dtype": "float32",
"dropout": 0.3
}