Update 'process_dataset.py'

This commit is contained in:
Kacper Dudzic 2022-04-02 15:56:52 +02:00
parent bf20f27daf
commit f148a3f24f

View File

@ -15,7 +15,7 @@ lego['prod_id'] = lego['prod_id'].apply(np.int64)
# wglad, statystyki # wglad, statystyki
print(lego) print(lego)
print(lego.describe(include='all')) print(lego.describe(include='all')).encode('utf-8')
# pierwszy podzial, wydzielamy zbior treningowy # 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, random_state=1)