From a063eea17b4f71bfc96ff430204cf37ebbaa0d1b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wojciech=20Miko=C5=82ajski?= Date: Fri, 16 Jun 2023 01:39:44 +0200 Subject: [PATCH] Update 'create_dataset.py' --- create_dataset.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/create_dataset.py b/create_dataset.py index 22d9dbf..068a6e4 100644 --- a/create_dataset.py +++ b/create_dataset.py @@ -49,5 +49,14 @@ water_test, water_dev = train_test_split(water_test, train_size=0.66, random_sta #water["is_safe"].value_counts().plot(kind="bar") -waterNorm.to_csv('waterQuality.csv', index=False) + + +directory ='dane' +if not os.path.exists(directory): + os.makedirs('dane') + +waterNorm.to_csv('waterQuality.csv', index=False) +water_train.to_csv('dane/diamonds_train.csv', index=False) +water_test.to_csv('dane/diamonds_test.csv', index=False) +water_dev.to_csv('dane/diamonds_dev.csv', index=False)