Skip to content

Commit

Permalink
Merge pull request #10 from godatadriven/dev
Browse files Browse the repository at this point in the history
Dev
  • Loading branch information
JasperHG90 authored Jan 16, 2021
2 parents 090b589 + a382489 commit eeaca13
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 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 All @@ -146,7 +146,7 @@ MyPivenModel(
Note that the inputs to `MyPivenModel` must match the inputs to the `piven_model` function.

You can now call all methods defined as in the PivenBaseModel class. Check the
[PivenMlpModel class](https://gitlab.com/jasperginn/piven.py/-/blob/dev/src/piven/Models/mlp_regressor.py)
[PivenMlpModel class](https://github.com/godatadriven/piven/blob/master/src/piven/models/mlp_regressor.py)
for a more detailed example.

## Details: loss function
Expand Down
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
[bumpversion]
current_version = 0.3.3
current_version = 0.4.0
commit = True
tag = True

Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

setuptools.setup(
name="piven",
version="0.3.3",
version="0.4.0",
description="Prediction Intervals with specific value prediction",
url="https://gitlab.com/jasperginn/piven.py",
author="Jasper Ginn",
Expand Down
2 changes: 1 addition & 1 deletion src/piven/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,4 @@

logging.getLogger(__name__).addHandler(logging.NullHandler())

__version__ = "0.3.3"
__version__ = "0.4.0"

0 comments on commit eeaca13

Please sign in to comment.