fixed probabilites operation between levels of classifier

alercebroker · Jul 23, 2021 · 4cca0ee · 4cca0ee
1 parent bac74c0
commit 4cca0ee
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lc_classifier/classifier/models.py b/lc_classifier/classifier/models.py
@@ -297,13 +297,13 @@ def predict_in_pipeline(self, input_features: pd.DataFrame) -> dict:
 
         prob_children = []
         resp_children = {}
-
         child_models = [
             self.stochastic_classifier,
-            self.periodic_classifier,
             self.transient_classifier,
+            self.periodic_classifier,
         ]
-        child_names = ["Stochastic", "Periodic", "Transient"]
+        child_names = ["Transient", "Stochastic", "Periodic"]
+
         for name, model in zip(child_names, child_models):
             prob_child = pd.DataFrame(
                 model.predict_proba(input_features),
@@ -312,7 +312,7 @@ def predict_in_pipeline(self, input_features: pd.DataFrame) -> dict:
             )
 
             resp_children[name] = prob_child
-            prob_child = prob_child.mul(prob_root[name].values, axis="rows")
+            prob_child = prob_child.mul(prob_root[name].values, axis=0)
             prob_children.append(prob_child)
         prob_all = pd.concat(prob_children, axis=1, sort=False)