b7eb836b40893bc25e782658cfec6f7bd6c4c451,seglearn/segment.py,Segment,transform,#Segment#Any#,26

Before Change

    def transform(self, X):
        check_is_fitted(self, "step")
        if type(X) is np.recarray:
            X_new = self._transform_recarray(X)
        else:
            X_new = self._transform(X)

        return X_new

    def _transform(self, X):
        N = len(X)

After Change

    def transform(self, X):
        check_is_fitted(self, "step")
        Xt = np.array([sliding_tensor(X["ts"][i], self.width, self.step) for i in range(len(X))])
        h_names = [h for h in X.dtype.names if h != "ts"]
        Xh = [X[h] for h in h_names]
        X_new = np.core.records.fromarrays([Xt] + Xh, names=["ts"] + h_names)
        return X_new

Instances