New Version

ashishpapanai · Nov 22, 2021 · e232e13 · e232e13
1 parent a16038b
commit e232e13
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 2 deletions.
diff --git a/lstm.h5 b/lstm.h5
diff --git a/mix_lstm.h5 b/mix_lstm.h5
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.4",
+    version="0.2.5",
     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.4"
+__version__ = "0.2.5"
 from . import calculations, data, main, market, models, plots, preprocessing, results, train
diff --git a/stockDL/train.py b/stockDL/train.py
@@ -68,4 +68,6 @@ def train_model(self):
         plt.show()
         self.models.lstm_model.save_weights("./lstm_weights.h5")
         self.models.mix_lstm_model.save_weights("./mix_lstm_weights.h5")
+        self.models.lstm_model.save("./lstm.h5")
+        self.models.mix_lstm_model.save("./mix_lstm.h5")
         return lstm_history, mix_history