Adrian8a commited on
Commit
ce9e383
1 Parent(s): 8225cb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -53,9 +53,11 @@ def getfig(X_test):
53
 
54
  fig = plt.figure(figsize=(5,4))
55
 
56
- plt.scatter(X_pca[:, 0], X_pca[:, 1], c = y, cmap = plt.cm.Spectral, s = 10)
57
  plt.scatter(u_pca[:, 0], u_pca[:, 1], c = 'g', cmap = plt.cm.Spectral, s = 40)
 
58
  plt.title(f"PCA, Exp. Variance: {np.round(np.sum(pca.explained_variance_ratio_), 4)}")
 
59
  plt.xlabel("PC 1")
60
  plt.ylabel("PC 2")
61
 
 
53
 
54
  fig = plt.figure(figsize=(5,4))
55
 
56
+ plt.scatter(X_pca[:, 0], X_pca[:, 1], c = y, cmap = plt.cm.Spectral, s = 10, label='More Chance AH')
57
  plt.scatter(u_pca[:, 0], u_pca[:, 1], c = 'g', cmap = plt.cm.Spectral, s = 40)
58
+ plt.annotate('You', (u_pca[:, 0]+10, u_pca[:, 1]+5), bbox = dict(boxstyle="round",ec='g',fc='lightgreen'))
59
  plt.title(f"PCA, Exp. Variance: {np.round(np.sum(pca.explained_variance_ratio_), 4)}")
60
+ plt.legend()
61
  plt.xlabel("PC 1")
62
  plt.ylabel("PC 2")
63