autism-detector / config.json
toderian's picture
Add config.json
023eb85 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.2,
"training": {
"learning_rate": 0.001,
"epochs": 300
}
}