s444501/ium_444501
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

quickfix

Browse Source
This commit is contained in:
s444501 2022-03-21 11:26:03 +01:00
parent 212b5fd3e5
commit 83b9cd31e3
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ium-data.py
View File

@ -42,6 +42,11 @@ disney['Review_Text'] = disney['Review_Text'].str.lower()
d_train, d_remainder = train_test_split(disney, test_size=0.2, random_state=1, stratify=disney["Branch"]) d_train, d_remainder = train_test_split(disney, test_size=0.2, random_state=1, stratify=disney["Branch"])
d_dev, d_test = train_test_split(d_remainder, test_size=0.5, random_state=1, stratify=d_remainder["Branch"]) d_dev, d_test = train_test_split(d_remainder, test_size=0.5, random_state=1, stratify=d_remainder["Branch"])
# Zapis do plików
d_train.to_csv('d_train.csv', index=False)
d_test.to_csv('d_test.csv', index=False)
d_dev.to_csv('d_dev.csv', index=False)
# Statystyki # Statystyki
print(f"Wielkość całego zbioru: {disney.shape[0]}\n" print(f"Wielkość całego zbioru: {disney.shape[0]}\n"
f"Inne statystyki:") f"Inne statystyki:")