diff --git a/process_dataset.py b/process_dataset.py index cf5f748..8a93890 100644 --- a/process_dataset.py +++ b/process_dataset.py @@ -14,8 +14,8 @@ lego['piece_count'] = lego['piece_count'].apply(np.int64) lego['prod_id'] = lego['prod_id'].apply(np.int64) # wglad, statystyki -print(lego[:10]) -print(lego.describe(include='all')[: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)