From 988ff371254d859f5d3ad6a4fa163ea50e09e625 Mon Sep 17 00:00:00 2001
From: suzusuzu <suzuzusu123@icloud.com>
Date: Fri, 27 Sep 2019 17:59:38 +0900
Subject: [PATCH] fix plot error

---
 GPclust/OMGP.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/GPclust/OMGP.py b/GPclust/OMGP.py
index d3ab358..3920925 100644
--- a/GPclust/OMGP.py
+++ b/GPclust/OMGP.py
@@ -204,7 +204,7 @@ def plot(self, gp_num=0):
         XX = np.linspace(self.X.min(), self.X.max())[:, None]
 
         if self.Y.shape[1] == 1:
-            plt.scatter(self.X, self.Y, c=self.phi[:, gp_num], cmap=cm.RdBu, vmin=0., vmax=1., lw=0.5)
+            plt.scatter(self.X[:, 0], self.Y[:, 0], c=self.phi[:, gp_num], cmap=cm.RdBu, vmin=0., vmax=1., lw=0.5)
             plt.colorbar(label='GP {} assignment probability'.format(gp_num))
 
             try: