Update 'train.py'

2023-09-30 00:50:34 +02:00 · 2023-09-30 00:50:34 +02:00 · b19a8e177d
commit b19a8e177d
parent 2d3022e622
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -27,7 +27,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('Index')
+    price_predict = train_data_x.pop('Price')
    normalize = layers.Normalization()
    normalize.adapt(price_train)