Make each AI task seperate for faster testing
This commit is contained in:
@@ -1,5 +1,4 @@
|
||||
import os
|
||||
import json
|
||||
import joblib
|
||||
import numpy as np
|
||||
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
||||
@@ -17,8 +16,10 @@ def Predict(Symbol):
|
||||
# Pull 1 month of current data to make prediction against | for volatility 20
|
||||
df = yf.download(Symbol, period="2mo", auto_adjust=True)
|
||||
if not df.empty:
|
||||
df = features.MakeFeatures(df, 1)
|
||||
df = features.CleanDF(df)
|
||||
# Remove the ticker column
|
||||
df.columns = df.columns.get_level_values(0)
|
||||
# Make the feature set
|
||||
df = features.MakeFeatures(df)
|
||||
|
||||
print(Symbol)
|
||||
|
||||
@@ -63,4 +64,4 @@ def Predict(Symbol):
|
||||
return movement_indicator
|
||||
|
||||
if __name__ == "__main__":
|
||||
Predict()
|
||||
Predict("AAPL")
|
||||
Reference in New Issue
Block a user