diff options
Diffstat (limited to 'Metrics/Metrics-Calculation/metrics_plot/src/test.py')
-rw-r--r-- | Metrics/Metrics-Calculation/metrics_plot/src/test.py | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/Metrics/Metrics-Calculation/metrics_plot/src/test.py b/Metrics/Metrics-Calculation/metrics_plot/src/test.py new file mode 100644 index 00000000..d1aae53a --- /dev/null +++ b/Metrics/Metrics-Calculation/metrics_plot/src/test.py | |||
@@ -0,0 +1,35 @@ | |||
1 | from sklearn.datasets import load_digits | ||
2 | from sklearn.manifold import MDS | ||
3 | import matplotlib.pyplot as plt | ||
4 | from scipy import stats | ||
5 | import numpy as np | ||
6 | |||
7 | dist = [] | ||
8 | |||
9 | for i in range(100): | ||
10 | rvs = stats.uniform.rvs(size=500, loc=0., scale=1) | ||
11 | dist.append(rvs) | ||
12 | |||
13 | for i in range(100): | ||
14 | rvs2 = stats.powerlaw .rvs(1.66, size=500) | ||
15 | dist.append(rvs2) | ||
16 | |||
17 | matrix = np.empty((len(dist),len(dist))) | ||
18 | |||
19 | for i in range(len(dist)): | ||
20 | matrix[i,i] = 0 | ||
21 | for j in range(i+1, len(dist)): | ||
22 | value, p = stats.ks_2samp(dist[i], dist[j]) | ||
23 | matrix[i, j] = value | ||
24 | matrix[j, i] = value | ||
25 | |||
26 | embedding = MDS(n_components=2, dissimilarity='precomputed') | ||
27 | trans = embedding.fit_transform(X=matrix) | ||
28 | x = (trans[:100,0]).tolist() | ||
29 | y = (trans[:100,1]).tolist() | ||
30 | |||
31 | x2 = (trans[100:,0]).tolist() | ||
32 | y2 = (trans[100:,1]).tolist() | ||
33 | plt.plot(x, y, 'yo') | ||
34 | plt.plot(x2, y2, 'ro') | ||
35 | plt.show() | ||