ComfyUI Node
Linear Regression Train
A ComfyUI node in d2l/Training with 7 inputs and 4 outputs.
Linear Regression Train
- X
- y
- dataset
- w
- b
- loss_history
- y_hat
◄num_steps100►
◄batch_size32►
◄lr0.030►
◄seed0►
Categoryd2l/Training
Inputs (7)
| Name | Type | Default | Description |
|---|---|---|---|
| X | TENSOR | — | |
| y | TENSOR | — | |
| num_steps | INT | 1001–100000 | — |
| batch_size | INT | 320–65536 | — |
| lr | FLOAT | 0.0300.00001–1 | — |
| seed | INT | 00–99999 | — |
| datasetopt | DATASET | — |
Outputs (4)
| Name | Type | Description |
|---|---|---|
| w | TENSOR | — |
| b | TENSOR | — |
| loss_history | TENSOR | — |
| y_hat | TENSOR | — |