fix a bug for one-dimensional input

yzhao062 · Apr 11, 2022 · f67f3ad · f67f3ad
1 parent ce50b69
commit f67f3ad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/suod/models/base.py b/suod/models/base.py
@@ -251,6 +251,10 @@ def fit(self, X):
         else:  # float
             self.target_dim_frac_ = int(self.target_dim_frac * n_features)
 
+        # it should have at least 1 dimension
+        if self.target_dim_frac_ < 1:
+            self.target_dim_frac_ = 1
+
         # build flags for random projection
         self.rp_flags_, _ = build_codes(self.base_estimators, self.rp_clf_list,
                                         self.rp_ng_clf_list,