From 1edea9e3d5d5158ac08640a80cd1636442e61b94 Mon Sep 17 00:00:00 2001 From: Jeong-Yoon Lee Date: Fri, 4 Feb 2022 16:29:11 -0800 Subject: [PATCH] switch the joblib backend for UpliftRandomForest to threading to optimize memory usage (#468) --- causalml/inference/tree/uplift.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/causalml/inference/tree/uplift.pyx b/causalml/inference/tree/uplift.pyx index fe33b783..3dbff39f 100644 --- 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) )