XGBoost - How to get the dataset