Make each AI task seperate for faster testing

This commit is contained in:
2026-02-26 21:39:41 -08:00
parent 79ee297e61
commit 7567496e1c
3 changed files with 13 additions and 32 deletions
+5 -4
View File
@@ -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")