8574d53937ee4bc0c657916fd2537af86d1a5db6,tslearn/shapelets.py,ShapeletModel,predict,#ShapeletModel#Any#,413

Before Change

        check_is_fitted(self, "X_fit_")
        X = check_array(X, allow_nd=True)

        if len(X.shape) == 2:
            warnings.warn("2-Dimensional data passed. Assuming these are "
                          "{} 1-dimensional timeseries".format(X.shape[0]))
            X = X.reshape((X.shape) + (1,))

        check_dims(self.X_fit_, X)
        categorical_preds = self.predict_proba(X)
        if self.categorical_y_:
            return categorical_preds
        else:

After Change

            what was provided at training time.
        
        check_is_fitted(self, "X_fit_")
        X = check_array(X, allow_nd=True)
        X = check_dims(X, X_fit=self.X_fit_)

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

Instances