ComfyUI Node
Regression Train
A ComfyUI node in d2l/Training with 15 inputs and 5 outputs.
Regression Train
- X
- y
- dataset
- nn_model
- mae
- rmse
- predictions
- loss_history
◄test_size0.20►
◄standardizeyes►
◄hidden►
◄activationrelu►
◄lossmse►
◄steps500►
◄batch_size0►
◄lr0.100►
◄seed0►
◄early_stop_patience30►
◄early_stop_min_delta0.0001►
◄save_path►
Categoryd2l/Training
Inputs (15)
| Name | Type | Default | Description |
|---|---|---|---|
| X | TENSOR | — | |
| y | TENSOR | — | |
| test_size | FLOAT | 0.200–0.95 | — |
| standardize | COMBO | yes | 2 options: yes, no |
| hidden | STRING | — | |
| activation | COMBO | relu | 4 options: relu, tanh, sigmoid, none |
| loss | COMBO | mse | 2 options: mse, mae |
| steps | INT | 5001–100000 | — |
| batch_size | INT | 00–65536 | — |
| lr | FLOAT | 0.1000.00001–1 | — |
| seed | INT | 00–99999 | — |
| early_stop_patience | INT | 300–100000 | — |
| early_stop_min_delta | FLOAT | 0.00010–1 | — |
| save_path | STRING | — | |
| datasetopt | DATASET | — |
Outputs (5)
| Name | Type | Description |
|---|---|---|
| nn_model | nn_model | — |
| mae | TENSOR | — |
| rmse | TENSOR | — |
| predictions | TENSOR | — |
| loss_history | TENSOR | — |