From 166c30997e851feceb50b354d936508315ade69f Mon Sep 17 00:00:00 2001 From: Kacper Dudzic Date: Mon, 9 May 2022 11:28:14 +0200 Subject: [PATCH] Update 'process_dataset.py' --- process_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/process_dataset.py b/process_dataset.py index 8a93890..2f54b7b 100644 --- a/process_dataset.py +++ b/process_dataset.py @@ -18,10 +18,10 @@ print(lego) print(lego.describe(include='all')) # pierwszy podzial, wydzielamy zbior treningowy -lego_train, lego_rem = train_test_split(lego, train_size=0.8, random_state=1) +lego_train, lego_rem = train_test_split(lego, train_size=0.8) # drugi podział, wydzielamy walidacyjny i testowy -lego_valid, lego_test = train_test_split(lego_rem, test_size=0.5, random_state=1) +lego_valid, lego_test = train_test_split(lego_rem, test_size=0.5) # zapis lego.to_csv('lego_sets_clean.csv', index=None, header=True)