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

asdasd

Browse Source
This commit is contained in:
s444501 2022-03-25 23:31:54 +01:00
parent 1483c3e08e
commit 1a610e8ce4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ium-data.py
View File

@ -39,8 +39,8 @@ disney['Review_Text'] = disney['Review_Text'].str.lower()
# Podział na podzbiory: d_train, d_test, d_dev
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_train, d_test = train_test_split(disney, test_size=0.2, random_state=1, stratify=disney["Branch"])
d_dev, d_test = train_test_split(d_test, test_size=0.5, random_state=1, stratify=d_test["Branch"])
# Zapis do plików
d_train.to_csv('d_train.csv', index=False)