Substituted investpy for yahoo finance

SCCapstone · Apr 26, 2021 · d79233b · d79233b
1 parent cfb989e
commit d79233b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/prediction/analysis.py b/prediction/analysis.py
@@ -4,7 +4,8 @@
 #import numpy as np
 #import keras
 import random
-import investpy as iv
+#import investpy as iv
+import yfinance as yf
 
 
 SAMPLE_P = 50
@@ -156,7 +157,9 @@ def get_rsi(percents, train=True):
 def get_prediction(ticker):
     upper = random.random()
     lower = random.random()
-    data = iv.get_stock_recent_data(stock=ticker, country='United States')
+    data = yf.Ticker(ticker).history(period="max")
+    #data = iv.get_stock_recent_data(stock=ticker, country='United States')
+    print(data)
     prediction = data['Open'][-1]
     upper_value = prediction * (1 + upper)
     lower_value = prediction * lower