From 7e16b7a80f3a2896351262a02af27a60401b6a5e Mon Sep 17 00:00:00 2001 From: Rémi Flamary Date: Thu, 3 Nov 2016 17:07:22 +0100 Subject: add mapping estimation with kernels (still debugging) --- ot/datasets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'ot/datasets.py') diff --git a/ot/datasets.py b/ot/datasets.py index 588f501..c750812 100644 --- a/ot/datasets.py +++ b/ot/datasets.py @@ -108,9 +108,9 @@ def get_data_classif(dataset,n,nz=.5,theta=0,**kwargs): x[y==3,:]+=2*nz*np.random.randn(sum(y==3),2) elif dataset.lower()=='gaussrot' : - rot=np.array([[np.cos(theta),-np.sin(theta)],[np.sin(theta),np.cos(theta)]]) - m1=np.array([-1,-1]) - m2=np.array([1,1]) + rot=np.array([[np.cos(theta),np.sin(theta)],[-np.sin(theta),np.cos(theta)]]) + m1=np.array([-1,1]) + m2=np.array([1,-1]) y=np.floor((np.arange(n)*1.0/n*2))+1 n1=np.sum(y==1) n2=np.sum(y==2) -- cgit v1.2.3