Switch from volitility to price tomorrow

2026-02-17 19:10:56 -08:00
parent b1a64be367
commit 7ff2f994e0
3 changed files with 8 additions and 5 deletions
@@ -16,8 +16,8 @@ def TrainAI():

    # Load the dataset
    dataset = pd.read_parquet(os.path.join(DATA_DIR, "stocks.parquet"))
-    X = dataset.drop('Volatility_5', axis=1)
-    Y = dataset['Volatility_5']
+    X = dataset.drop('Target_Close_Tomorrow', axis=1)
+    Y = dataset['Target_Close_Tomorrow']

    # Show the datatypes
    print(dataset.dtypes)