File size: 3,358 Bytes
93fc950
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
{
  "setting": {
    "name": {
      "text": "Nome do modelo",
      "value": "mpl_pequeno_torch",
      "Help": "Identificação do modelo PyTorch tabular. Pode ser usado para salvar checkpoints ou registros.",
      "default": "mpl_pequeno_torch",
      "type": "string",
      "required": true
    },
    "architecture": {
      "text": "Arquitetura do modelo",
      "value": "MLP tabular",
      "Help": "Arquitetura da rede neural para dados tabulares usando PyTorch.",
      "default": "MLP tabular",
      "type": "string",
      "required": true
    },
    "tipo": {
      "text": "Tipo do modelo",
      "value": "classification",
      "Help": "Define o tipo de tarefa do modelo.",
      "default": "classification",
      "type": "string",
      "required": true
    },
      "learning_type": {
      "text": "Tipo de aprendizado",
      "value": "deep_learning",
      "Help": "Indica se o modelo é de machine learning clássico ou deep learning.",
      "default": "deep_learning",
      "type": "string",
      "required": true
    }
  },
  "train": {
    "batch_size": {
      "text": "Tamanho do batch",
      "Help": "Tamanho do batch usado durante o treinamento.",
      "default": 32,
      "range": {
        "min": 1,
        "max": null
      },
      "type": "integer",
      "required": true
    },
    "epochs": {
      "text": "Número de épocas",
      "Help": "Quantidade de épocas de treinamento.",
      "default": 10,
      "range": {
        "min": 1,
        "max": 1000
      },
      "type": "integer",
      "required": true
    },
    "learning_rate": {
      "text": "Taxa de aprendizado",
      "Help": "Taxa usada pelo otimizador.",
      "default": 0.001,
      "range": {
        "min": 0.000001,
        "max": 1.0
      },
      "type": "float",
      "required": true
    },
    "weight_decay": {
      "text": "Decaimento de peso",
      "Help": "Regularização L2 usada pelo otimizador.",
      "default": 0.0001,
      "range": {
        "min": 0.0,
        "max": 0.1
      },
      "type": "float",
      "required": false
    },
    "optimizer": {
      "text": "Otimizador",
      "Help": "Otimizador usado no treinamento.",
      "default": "adam",
      "type": "string",
      "required": true
    },
    "hidden_dims": {
      "text": "Dimensões ocultas",
      "Help": "Lista de camadas ocultas do MLP.",
      "default": [32, 16],
      "type": "list",
      "required": false
    },
    "dropout": {
      "text": "Dropout",
      "Help": "Taxa de dropout aplicada entre camadas.",
      "default": 0.0,
      "range": {
        "min": 0.0,
        "max": 1.0
      },
      "type": "float",
      "required": false
    }
  },
  "test": {
    "batch_size": {
      "text": "Tamanho do batch",
      "Help": "Tamanho do batch usado na avaliação.",
      "default": 64,
      "range": {
        "min": 1,
        "max": null
      },
      "type": "integer",
      "required": true
    }
  },
  "predict": {
    "batch_size": {
      "text": "Tamanho do batch",
      "Help": "Tamanho do batch usado na predição.",
      "default": 64,
      "range": {
        "min": 1,
        "max": null
      },
      "type": "integer",
      "required": true
    }
  }
}