From 1a610e8ce4b3f04e73fb5543eb028f7303b02300 Mon Sep 17 00:00:00 2001 From: s444501 Date: Fri, 25 Mar 2022 23:31:54 +0100 Subject: [PATCH] asdasd --- ium-data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ium-data.py b/ium-data.py index d5ad1a4..50dfc21 100755 --- a/ium-data.py +++ b/ium-data.py @@ -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)