Update 'train.py'

This commit is contained in:
Tomasz Koszarek 2023-09-30 00:52:52 +02:00
parent b19a8e177d
commit 1b6c86630f

View File

@ -24,7 +24,7 @@ def my_config():
def main(EPOCHS, _run):
if EPOCHS == 0:
EPOCHS = 500
train_data_x = pd.read_csv('./train_data.csv')
train_data_x = pd.read_csv('./train_data2.csv')
price_train = train_data_x.copy()
price_predict = train_data_x.pop('Price')