Benjamin Bossan commited on
Commit
9be4332
1 Parent(s): 5444690

Add spiral dataset

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -88,10 +88,24 @@ def get_varied(n_clusters):
88
  return normalize(X), labels
89
 
90
 
 
 
 
 
 
 
 
 
 
 
 
 
 
91
  DATA_MAPPING = {
92
  'regular': get_regular,
93
  'circles': get_circles,
94
  'moons': get_moons,
 
95
  'noise': get_noise,
96
  'anisotropic': get_anisotropic,
97
  'varied': get_varied,
 
88
  return normalize(X), labels
89
 
90
 
91
+ def get_spiral(n_clusters):
92
+ # from https://scikit-learn.org/stable/auto_examples/cluster/plot_agglomerative_clustering.html
93
+ t = 1.5 * np.pi * (1 + 3 * np.random.rand(1, N_SAMPLES))
94
+ x = t * np.cos(t)
95
+ y = t * np.sin(t)
96
+ X = np.concatenate((x, y))
97
+ X += 0.7 * np.random.randn(2, N_SAMPLES)
98
+ X = X.T
99
+
100
+ labels = np.zeros(N_SAMPLES, dtype=int)
101
+ return normalize(X), labels
102
+
103
+
104
  DATA_MAPPING = {
105
  'regular': get_regular,
106
  'circles': get_circles,
107
  'moons': get_moons,
108
+ 'spiral': get_spiral,
109
  'noise': get_noise,
110
  'anisotropic': get_anisotropic,
111
  'varied': get_varied,