Predict class outputs for an unlabelled feature set

def predict(self, features):
        """Predict class outputs for an unlabelled feature set"""

        # get distance of features to class clusters
        distances = [self._distance(x) for x in features]

        # assign class label belonging to smallest distance
        class_predict = [np.argmin(d) for d in distances]

        return self.le.inverse_transform(class_predict)