fixes

ashishpapanai · Aug 28, 2021 · 154c170 · 154c170
1 parent 80bbd84
commit 154c170
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 # This call to setup() does all the work
 setup(
     name="stockDL",
-    version="0.2.2",
+    version="0.2.3",
     description="Predicts the Gross Yield, Annual Yield and Net Yield of a user given stock ticker.",
     long_description=README,
     long_description_content_type="text/markdown",

diff --git a/stockDL/__init__.py b/stockDL/__init__.py
@@ -1,3 +1,3 @@
 ''' Version of module '''
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 from . import calculations, data, main, market, models, plots, preprocessing, results, train
diff --git a/stockDL/train.py b/stockDL/train.py
@@ -60,6 +60,6 @@ def train_model(self):
                                                      batch_size=48, validation_data=(self.preprocessing.X_test,
                                                                                      self.preprocessing.y_test),
                                                      verbose=2, callbacks=[self.learning_rate_reduction], shuffle=False)
-        self.models.lstm_model.save_weights("./stockDL/lstm_weights.h5")
-        self.models.mix_lstm_model.save_weights("./stockDL/mix_lstm_weights.h5")
+        self.models.lstm_model.save_weights("./lstm_weights.h5")
+        self.models.mix_lstm_model.save_weights("./mix_lstm_weights.h5")
         return lstm_history, mix_history