Update 'process_dataset.py'

2022-05-09 11:28:14 +02:00 · 2022-05-09 11:28:14 +02:00 · 166c30997e
commit 166c30997e
parent 674c95103e
1 changed files with 2 additions and 2 deletions
--- 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)