diff --git a/lab5/create/create_dataset.py b/lab5/create/create_dataset.py index fc43ba8..04f8c26 100644 --- a/lab5/create/create_dataset.py +++ b/lab5/create/create_dataset.py @@ -30,7 +30,7 @@ train = pd.concat([X_train, Y_train], axis=1) test = pd.concat([X_test, Y_test], axis=1) valid = pd.concat([X_valid, Y_valid], axis=1) -train.to_csv('train.csv', index=False) test.to_csv('test.csv', index=False) valid.to_csv('valid.csv', index=False) +train.to_csv('train.csv', index=False)