GridSearchTraining
GridSearchTraining
Pack: EternalKernel PyTorch Nodes
custom_nodes.EternalKernelPytorchNodes
Inputs (5)
| Name | Type | Required |
|---|---|---|
| model | TORCH_MODEL | required |
| param_grid | STRING | required |
| dataset | TORCH_DATASET | optional |
| features tensor | TORCH_TENSOR | optional |
| labels tensor | TORCH_TENSOR | optional |
Outputs (3)
| Name | Type |
|---|---|
| best_model | TORCH_MODEL |
| best_params | DICT |
| all_metrics | LIST |