Skip to content

Commit

Permalink
docs: update method name
Browse files Browse the repository at this point in the history
  • Loading branch information
JasperHG90 committed Jan 16, 2021
1 parent 37f6ca4 commit 8657da5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def piven_model(input_size, hidden_units):
```

The most straightforward way of running your Model is to subclass the `PivenBaseModel` class. This requires you
to define a `build_model()` method in which you can add preprocessing pipelines etc.
to define a `build()` method in which you can add preprocessing pipelines etc.

```python
from piven.models.base import PivenBaseModel
Expand All @@ -125,7 +125,7 @@ from sklearn.preprocessing import StandardScaler


class MyPivenModel(PivenBaseModel):
def build_model(self, build_fn = piven_model):
def build(self, build_fn = piven_model):
model = PivenKerasRegressor(build_fn=build_fn, **self.params)
# Finally, normalize the output target
self.model = PivenTransformedTargetRegressor(
Expand Down

0 comments on commit 8657da5

Please sign in to comment.