Rollback cond gaus (aimclub#92)

* serialization fix * min_shape changed
jrzkaminski · Nov 30, 2023 · 1315151 · 1315151
1 parent b1637a3
commit 1315151
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bamt/nodes/conditional_gaussian_node.py b/bamt/nodes/conditional_gaussian_node.py
@@ -44,7 +44,7 @@ def fit_parameters(self, data: DataFrame) -> Dict[str, Dict[str, CondGaussParams
                 mask = (mask) & (data[col] == val)
             new_data = data[mask]
             key_comb = [str(x) for x in comb]
-            if new_data.shape[0] > 1:
+            if new_data.shape[0] > 0:
                 if self.cont_parents:
                     model = clone(self.regressor)
                     model.fit(