eb4212aff509af724c5cc011a72c5d181ae12ae4,tslearn/shapelets.py,ShapeletModel,predict,#ShapeletModel#Any#,291

Before Change

        if self.categorical_y:
            return categorical_preds
        else:
            return categorical_preds.argmax(axis=1)

    def transform(self, X):
        Generate shapelet transform for a set of time series.

After Change

        if self.categorical_y:
            return categorical_preds
        else:
            return self.label_binarizer.inverse_transform(categorical_preds)

    def transform(self, X):
        Generate shapelet transform for a set of time series.

Instances