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
}
}
} |