fix enf of file
This commit is contained in:
parent
eabd321281
commit
8903ba5d89
@ -325,4 +325,4 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
|
||||
return trainer
|
||||
```
|
||||
|
||||
Here we create `PyTorchMLPRegressor` that implements the `fit` method. The `fit` method specifies the training building blocks: model, optimizer, criterion, and trainer. We inherit both `BasePyTorchRegressor` and `BasePyTorchModel` (a parent of `BasePyTorchRegressor`). The former implements the `predict` method that suits our regression task. The latter implements the `train` method.
|
||||
Here we create `PyTorchMLPRegressor` that implements the `fit` method. The `fit` method specifies the training building blocks: model, optimizer, criterion, and trainer. We inherit both `BasePyTorchRegressor` and `BasePyTorchModel` (a parent of `BasePyTorchRegressor`). The former implements the `predict` method that suits our regression task. The latter implements the `train` method.
|
||||
|
Loading…
Reference in New Issue
Block a user