9a00ddb56e2f3e1e0142f5c3675fc6d9ea7ec017,dbn/tensorflow/models.py,SupervisedDBNClassification,_transform_network_format_to_labels,#SupervisedDBNClassification#,454

Before Change

        :param indexes: array-like, shape = (n_samples, )
        :return:
        
        return map(lambda idx: self.idx_to_label_map[idx], indexes)

    def predict(self, X):
        probs = self.predict_proba(X)
        indexes = np.argmax(probs, axis=1)

After Change

        :param indexes: array-like, shape = (n_samples, )
        :return:
        
        return list(map(lambda idx: self.idx_to_label_map[idx], indexes))

    def predict(self, X):
        probs = self.predict_proba(X)
        indexes = np.argmax(probs, axis=1)

Instances