046eb11cc615c3232b93821c78895152bad2159a,examples/ERP/plot_embedding_EEG.py,,,#,24

Before Change

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Embedding the Xdawn covariance matrices with Diffusion maps

covs = ERPCovariances(estimator="oas", classes=[3, 4]).fit_transform(X, y)
lapl = Embedding(metric="riemann", n_components=3)
embd = lapl.fit_transform(covs)

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

After Change

xdwn = XdawnCovariances(estimator="oas", nfilter=nfilter)
split = train_test_split(X, y, train_size=0.5, random_state=42)
Xtrain, Xtest, ytrain, ytest = split
covs = xdwn.fit(Xtrain, ytrain).transform(Xtest)

lapl = Embedding(metric="riemann", n_components=2)
embd = lapl.fit_transform(covs)

Instances