diff --git a/train.py b/train.py
index 0d62dba..70b6a79 100644
--- a/train.py
+++ b/train.py
@@ -25,7 +25,7 @@ def my_config():
 def my_main(EPOCHS, _run):
     _run.info["epochs"] = EPOCHS,
     normalize = layers.Normalization()
-    train_data_x = pandas.read_csv('./data_train.csv')
+    train_data_x = pandas.read_csv('./train_data.csv')
     _run.info["dataset"] = train_data_x
     dat_all = train_data_x.copy()
     dat_predict = train_data_x.pop('Price')