Update 'train.py'

2023-09-30 00:37:11 +02:00 · 2023-09-30 00:37:11 +02:00 · f13a2c8603
commit f13a2c8603
parent d9f0fa862a
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -26,7 +26,7 @@ def main(EPOCHS, _run):
    train_data_x = pd.read_csv('./train_data.csv')

    price_train = train_data_x.copy()
-    price_predict = train_data_x.pop('Price')
+    price_predict = train_data_x.pop('Index')
    normalize = layers.Normalization()
    normalize.adapt(price_train)