patches to deal with categorical data in xgboost

loft-br · Jul 1, 2022 · 078402e · 078402e
1 parent 60fce5b
commit 078402e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xgbse/_kaplan_neighbors.py b/xgbse/_kaplan_neighbors.py
@@ -158,6 +158,7 @@ def fit(
         if time_bins is None:
             time_bins = get_time_bins(self.T_train, self.E_train)
         self.time_bins = time_bins
+        # TODO check this section....
 
         # converting data to xgb format
         dtrain = convert_data_to_xgb_format(X, y, self.xgb_params["objective"], enable_categorical=enable_categorical)