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

Add 'dataset-stats.py'

Browse Source
This commit is contained in:
Wojciech Mikołajski 2023-04-18 19:55:09 +02:00
parent 8a7c05a5d8
commit 6850eec9d6
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
dataset-stats.py Normal file
View File

@ -0,0 +1,22 @@
import pandas as pd
import os
import sys
from sklearn.model_selection import train_test_split
water = pd.read_csv('waterQuality.csv')
water_train, water_test = train_test_split(water, train_size=0.8, random_state=1, stratify=water["is_safe"])
water_test, water_dev = train_test_split(water_test, train_size=0.66, random_state=1, stratify=water_test["is_safe"])
water_train["is_safe"].value_counts()
water_test["is_safe"].value_counts()
water_dev["is_safe"].value_counts()
print(f'''
Statystyki zbioru:
Wielkość zbioru - {len(water)}
Wielkość podzbioru treningowego - {len(water_train)}
Wielkość podzbioru walidującego - {len(water_dev)}
Wielkość podzbioru testowego - {len(water_test)}
Rozkład częstości parametru mówiącemu o zdantości picia wody (0 oznacza zdanty do picia):
''')