Skip to content
This repository has been archived by the owner on May 10, 2024. It is now read-only.

Fixed compute of probabilities

Compare
Choose a tag to compare
@JavierArredondo JavierArredondo released this 23 Jul 16:11

Fixed order of operations in compute of probabilities (models.py):

        prob_children = []
        resp_children = {}
        child_models = [
            self.stochastic_classifier,
            self.transient_classifier,
            self.periodic_classifier,
        ]
        child_names = ["Transient", "Stochastic", "Periodic"]

        for name, model in zip(child_names, child_models):
            prob_child = pd.DataFrame(
                model.predict_proba(input_features),
                columns=model.classes_,
                index=input_features.index,
            )

            resp_children[name] = prob_child
            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)