Update xgboostlss.py

sktime · Jan 25, 2025 · d7c5dac · d7c5dac
1 parent 2d64c5c
commit d7c5dac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/skpro/regression/xgboostlss.py b/skpro/regression/xgboostlss.py
@@ -129,7 +129,7 @@ def _fit(self, X, y):
         self._y_cols = y.columns
         n_cpu = self._n_cpu
 
-        dtrain = xgb.DMatrix(X, label=y, n_cpu=n_cpu, silent=True)
+        dtrain = xgb.DMatrix(X, label=y, n_thread=n_cpu, silent=True)
 
         xgblss = XGBoostLSS(
             Gaussian(
@@ -211,7 +211,7 @@ def _predict_proba(self, X):
         y_cols = self._y_cols
         columns = y_cols
 
-        dtest = xgb.DMatrix(X, n_cpu=n_cpu, silent=True)
+        dtest = xgb.DMatrix(X, n_thread=n_cpu, silent=True)
 
         y_pred_xgblss = self.xgblss_.predict(dtest, pred_type="parameters")