9302d6f56cb836b95e357ed533a3d3aaed4f4016,pyriemann/tangentspace.py,FGDA,fit_transform,#FGDA#Any#Any#,58

Before Change

       return cov

    def fit_transform(self,X,y=None):
       self.classes= numpy.unique(y)
       self._lda = LDA(n_components=len(self.classes)-1,solver="lsqr", shrinkage="auto")
       
       ts = self._ts.fit_transform(X)
       self._lda.fit(ts,y)
       
       W = self._lda.coef_
       self._W = numpy.dot(numpy.dot(W.T,numpy.linalg.pinv(numpy.dot(W,W.T))),W)
       ts = numpy.dot(ts,self._W)
       cov = self._ts.inverse_transform(ts)
       return cov

After Change

    def fit_transform(self,X,y=None):
        ts = self._fit_lda(X,y)
        return self._retro_project(ts)

Instances