File size: 426 Bytes
4862a83
 
 
 
023eb85
 
 
 
4862a83
 
 
 
 
 
 
 
 
 
 
023eb85
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
{
  "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
  }
}