s487182/ium_s487182
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'create_dataset.py'

Browse Source
This commit is contained in:
Wojciech Mikołajski 2023-06-16 01:39:44 +02:00
parent 889568384d
commit a063eea17b
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
create_dataset.py
View File

@ -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)