85aabb5014e22659ac722280607a1f4b44e1fb32,tslearn/shapelets.py,ShapeletModel,predict,#ShapeletModel#Any#,452

Before Change

        X = check_dims(X, X_fit_dims=self._X_fit_dims)

        categorical_preds = self.predict_proba(X)
        if self.categorical_y_:
            return categorical_preds
        else:
            return self.label_binarizer_.inverse_transform(categorical_preds)

    def predict_proba(self, X):
        Predict class probability for a given set of time series.

        Parameters

After Change

        y_label = numpy.array(
            [self.classes_[ind] for ind in y_ind]
        )
        return y_label

    def predict_proba(self, X):
        Predict class probability for a given set of time series.

Instances