diff --git a/setup.py b/setup.py index e1d09de..4d71cc9 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setuptools.setup( name="keras-beats-jonathan-bechtel", - version="0.0.2", + version="0.0.3", author="Jonathan Bechtel", author_email="jonathan@jonathanbech.tel", description="Lightweight installation of NBeats NN architecture for keras", diff --git a/src/kerasbeats/__init__.py b/src/kerasbeats/__init__.py new file mode 100644 index 0000000..9590337 --- /dev/null +++ b/src/kerasbeats/__init__.py @@ -0,0 +1 @@ +from . import GenericBlock, TrendBlock, SeasonalBlock, NBeats