diff --git a/multimodal/boosting/cumbo.py b/multimodal/boosting/cumbo.py index b498064cc18f703c9afc7d50811c947f87585651..158c9cacdd07753fa7c1149ef43e061e0b8c631d 100644 --- a/multimodal/boosting/cumbo.py +++ b/multimodal/boosting/cumbo.py @@ -216,10 +216,10 @@ class MuCumboClassifier(BaseEnsemble, ClassifierMixin, UBoosting): """Compute values of confidence rate alpha given edge values.""" np.where(edges > 1.0, edges, 1.0) alphas = 0.5 * np.log((1. + edges) / (1. - edges)) - if np.any(np.isinf(alphas)) or np.any(np.isnan(alphas)): - - alphas[np.where(np.isnan(alphas))[0]] = 1.0 + if np.any(np.isinf(alphas)): alphas[np.where(np.isinf(alphas))[0]] = 1.0 + if np.any(np.isnan(alphas)): + alphas[np.where(np.isnan(alphas))[0]] = 1.0 return alphas def _compute_cost(self, label_score, predicted_classes, y, alphas, betas, diff --git a/multimodal/boosting/mumbo.py b/multimodal/boosting/mumbo.py index 84530d8fd8551aabb7836a293e659148cac44bf1..23dacffa077ead81160b5571352e38a4c9438719 100644 --- a/multimodal/boosting/mumbo.py +++ b/multimodal/boosting/mumbo.py @@ -223,10 +223,10 @@ class MumboClassifier(BaseEnsemble, ClassifierMixin, UBoosting): """Compute values of confidence rate alpha given edge values.""" np.where(edges > 1.0, edges, 1.0) alphas = 0.5 * np.log((1. + edges) / (1. - edges)) - if np.any(np.isinf(alphas)) or np.any(np.isnan(alphas)): - - alphas[np.where(np.isnan(alphas))[0]] = 1.0 + if np.any(np.isinf(alphas)): alphas[np.where(np.isinf(alphas))[0]] = 1.0 + if np.any(np.isnan(alphas)): + alphas[np.where(np.isnan(alphas))[0]] = 1.0 return alphas def _compute_cost_global(self, label_score_global, best_predicted_classes,