f7a04171e58eb43dfe5b18d06c76481cdf1c5da9,deslib/dcs/lca.py,LCA,estimate_competence,#LCA#,85

Before Change

                    if self.DSEL_target[index] == predicted_label:
                        result.append(self.processed_dsel[index][clf_index])

                if len(result) == 0:
                    competences[clf_index] = 0.0
                else:
                    competences[clf_index] = np.mean(result)

        return competences

After Change

        // Expanding the dimensions of the predictions and target arrays in order to compare both.
        predictions_3d = np.expand_dims(predictions, axis=1)
        target_3d = np.expand_dims(self.DSEL_target[idx_neighbors], axis=2)
        // Create a mask to remove the neighbors belonging to a different class than the predicted by the base classifier
        mask = (predictions_3d != target_3d)
        masked_preprocessed = np.ma.MaskedArray(self.processed_dsel[idx_neighbors, :], mask=mask)

        competences_masked = np.mean(masked_preprocessed, axis=1)
        // Fill 0 to the masked values in the resulting array (when no neighbors belongs to the class predicted by
        // the corresponding base classifier)
        competences = np.ma.filled(competences_masked, 0)

Instances