Merge branch 'dev' into postmodeling_revision_2023

dssg · Nov 29, 2023 · faaa5de · faaa5de
2 parents 8b147ed + b9c4d0a
commit faaa5de
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/triage/component/catwalk/__init__.py b/src/triage/component/catwalk/__init__.py
@@ -47,6 +47,7 @@ def __init__(
         self.replace = replace
         self.protected_groups_generator = protected_groups_generator
         self.bigtrain_classnames = [
+            'imblearn.ensemble.BalancedRandomForestClassifier',
             'sklearn.ensemble.RandomForestClassifier',
             'sklearn.ensemble.ExtraTreesClassifier',
             'sklearn.ensemble.AdaBoostClassifier',