SYNV2

RodneyFinkel · Sep 25, 2023 · d0d4ff6 · d0d4ff6
1 parent 8ad2bb5
commit d0d4ff6
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/__pycache__/stock_screener.cpython-311.pyc b/__pycache__/stock_screener.cpython-311.pyc
diff --git a/app.py b/app.py
@@ -19,7 +19,7 @@
     company = cells[1].text.strip()
     sector = cells[3].text.strip()
     sp500.append({'ticker': ticker, 'company': company, 'sector': sector})
-    pprint(sp500)
+
 
 
 def get_sp500_stocks():

diff --git a/stock_screener.py b/stock_screener.py
@@ -284,7 +284,7 @@ def train_models(self):
 
             #Create and train model
             model = create_model(train_data) 
-            model.fit(train_data, train_labels, epochs=10)
+            model.fit(train_data, train_labels, epochs=20)
             self.models[stock.ticker] = model # models needs to be defined as a StockScreener attribute, namely a dictionary
 
     # Predict whether new stocks will pass filters (new_stocks gets passed as filtered_stocks in app.py)