cuda docs

sgoldenlab · Sep 3, 2024 · 2b2b131 · 2b2b131
1 parent c057c87
commit 2b2b131
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/simba/mixins/train_model_mixin.py b/simba/mixins/train_model_mixin.py
@@ -681,7 +681,6 @@ def create_x_importance_log(self,
             importances = list(self.cuml_rf_x_importances(nodes=cuml_tree_nodes, n_features=len(x_names)))
             std_importances = [np.nan] * len(importances)
         else:
-            print('s')
             importances_per_tree = np.array([tree.feature_importances_ for tree in rf_clf.estimators_])
             importances = list(np.mean(importances_per_tree, axis=0))
             std_importances = list(np.std(importances_per_tree, axis=0))