switch the joblib backend for UpliftRandomForest to threading to opti…

…mize memory usage (#468)
uber · Feb 5, 2022 · 1edea9e · 1edea9e
1 parent 313b4c4
commit 1edea9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/causalml/inference/tree/uplift.pyx b/causalml/inference/tree/uplift.pyx
@@ -1321,7 +1321,7 @@ class UpliftRandomForestClassifier:
         self.n_class = len(self.classes_)
 
         self.uplift_forest = (
-            Parallel(n_jobs=self.n_jobs)
+            Parallel(n_jobs=self.n_jobs, backend="threading")
             (delayed(self.bootstrap)(X, treatment, y, tree) for tree in self.uplift_forest)
         )