This repository has been archived by the owner on May 10, 2024. It is now read-only.
Fixed compute of probabilities
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)