vumichien commited on
Commit
03250df
·
1 Parent(s): b274e21

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +232 -0
app.py ADDED
@@ -0,0 +1,232 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+ import plotly.graph_objects as go
3
+ import matplotlib.pyplot as plt
4
+
5
+ from sklearn.datasets import load_digits
6
+ from sklearn.preprocessing import MinMaxScaler
7
+
8
+ from sklearn.decomposition import TruncatedSVD
9
+ from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
10
+ from sklearn.ensemble import RandomTreesEmbedding
11
+ from sklearn.manifold import (
12
+ Isomap,
13
+ LocallyLinearEmbedding,
14
+ MDS,
15
+ SpectralEmbedding,
16
+ TSNE
17
+ )
18
+ from sklearn.neighbors import NeighborhoodComponentsAnalysis
19
+ from sklearn.pipeline import make_pipeline
20
+
21
+ import gradio as gr
22
+
23
+ #====================================================================
24
+
25
+ digits = load_digits(n_class=10)
26
+ X, y = digits.data, digits.target
27
+
28
+ algorithms = ['Linear Discriminant Analysis',
29
+ 't-SNE',
30
+ 'Truncated SVD',
31
+ 'Isomap',
32
+ 'Locally Linear Embedding',
33
+ 'Random Trees Embedding',
34
+ 'Spectral Embedding',
35
+ 'Neighborhood Component Analysis']
36
+
37
+ #====================================================================
38
+ class UserInterface:
39
+ def __init__(self):
40
+ self.components = {}
41
+ self.bindings = {}
42
+
43
+ def add_component(self, name, component, kwargs={}):
44
+ self.components[name] = component(**{'visible': False, **kwargs})
45
+
46
+ def all_components(self):
47
+ return list(self.components.values())
48
+
49
+ def get(self, name):
50
+ return self.components[name]
51
+
52
+ def get_kwargs(self, args):
53
+ return {k:v for k,v in zip(list(self.components.keys()), args)}
54
+
55
+ def get_arg(self, name, args):
56
+ idx = list(self.components.keys()).index(name)
57
+ return args[idx]
58
+
59
+ def add_updater(self, dest, srcs, conds):
60
+ def fn(*args):
61
+ truth_value = True
62
+ for (src, cond) in zip(args, conds):
63
+ truth_value = truth_value and (src == cond)
64
+
65
+ return gr.update(visible=truth_value)
66
+
67
+ for src in srcs:
68
+ self.components[src].change(fn, inputs=[self.components[i] for i in srcs], outputs=self.components[dest])
69
+
70
+ def bind(self, dest, src, cond):
71
+ all_srcs, all_conds = [src], [cond]
72
+
73
+ if src in self.bindings:
74
+ for sub_src, sub_cond in self.bindings[src]:
75
+ all_srcs.append(sub_src)
76
+ all_conds.append(sub_cond)
77
+
78
+ self.add_updater(dest, all_srcs, all_conds)
79
+
80
+ if dest not in self.bindings: self.bindings[dest] = []
81
+ for s, c in zip(all_srcs, all_conds):
82
+ self.bindings[dest].append((s, c))
83
+
84
+
85
+ info = '''
86
+ # Giảm kích thước MNIST với học tập đa dạng
87
+
88
+ Đây là minh chứng về cách có thể sử dụng một số kỹ thuật [học đa dạng](https://scikit-learn.org/stable/modules/manifold.html) để giảm số chiều của tập dữ liệu MNIST từ 784 chiều xuống chỉ còn 2 hoặc 3 chiều cho mục đích trực quan.
89
+
90
+ Các kỹ thuật học đa dạng tìm cách tìm ra **biểu diễn chiều thấp** của dữ liệu đã cho sao cho các đặc tính nhất định được bảo toàn, chẳng hạn như cấu trúc cục bộ hoặc toàn cầu, khoảng cách lân cận, v.v.
91
+
92
+ Tất cả các phương pháp được sử dụng ở đây đều **hoàn toàn không được giám sát**, ngoại trừ Phân tích phân biệt tuyến tính và Phân tích thành phần lân cận,
93
+
94
+ sử dụng nhãn chữ số được cung cấp. Mỗi phương pháp đều có ưu điểm và nhược điểm. Sử dụng hộp thả xuống bên dưới để chọn một phương pháp, điều chỉnh các siêu tham số của chúng và xem chúng khác nhau như thế nào.
95
+ '''
96
+
97
+ with gr.Blocks(analytics_enabled=False,theme=gr.themes.Soft()) as demo:
98
+ gr.Markdown(info)
99
+
100
+ with gr.Row():
101
+ with gr.Column():
102
+ ui = UserInterface()
103
+ ui.add_component('algorithm', gr.Dropdown, {'choices': algorithms, 'value': algorithms[0], 'label': 'Model', 'interactive': True, 'visible': True})
104
+ ui.add_component('n_components', gr.Radio, {'choices': ['2D', '3D'], 'value': '3D', 'label': 'Embedding dimensionality', 'visible': True})
105
+
106
+ ui.add_component('n_neighbors', gr.Slider, {'minimum': 6, 'maximum': 100, 'value': 30, 'step': 1, 'label': 'Number of neighbors'})
107
+
108
+ ui.add_component('tsne_perplexity', gr.Slider, {'minimum': 5, 'maximum': 50, 'value': 30, 'step': 1, 'label': 'Perplexity'})
109
+ ui.add_component('tsne_early_exaggeration', gr.Slider, {'minimum': 2, 'maximum': 100, 'value': 12, 'step': 1, 'label': 'Early Exaggeration'})
110
+ ui.add_component('tsne_n_iter', gr.Slider, {'minimum': 250, 'maximum': 2000, 'value': 1000, 'step': 1, 'label': 'Number of iterations'})
111
+ ui.add_component('tsne_learning_rate_dd', gr.Dropdown, {'choices': ['auto', 'float'], 'value': 'auto', 'label': 'Learning Rate', 'interactive': True})
112
+ ui.add_component('tsne_learning_rate_s', gr.Slider, {'minimum': 10, 'maximum': 2000, 'value': 100, 'step': 1, 'label': 'Learning Rate Value'})
113
+
114
+ ui.add_component('isomap_p', gr.Slider, {'minimum': 1, 'maximum': 10, 'value': 2, 'step': 1, 'label': 'Minkowski distance Lp'})
115
+
116
+ ui.add_component('lle_method', gr.Dropdown, {'choices': ['standard', 'hessian', 'modified', 'ltsa'], 'value': 'standard', 'label': 'Method', 'interactive': True})
117
+
118
+ ui.add_component('rte_n_estimators', gr.Slider, {'minimum': 1, 'maximum': 200, 'value': 100, 'step': 1, 'label': 'Number of estimators'})
119
+ ui.add_component('rte_max_depth', gr.Slider, {'minimum': 2, 'maximum': 50, 'value': 5, 'step': 1, 'label': 'Max depth'})
120
+
121
+ ui.add_component('se_affinity', gr.Dropdown, {'choices': ['rbf', 'nearest_neighbors'], 'value': 'nearest_neighbors', 'label': 'Affinity matrix calculation method', 'interactive': True})
122
+ ui.add_component('se_gamma', gr.Slider, {'minimum': 0.01, 'maximum': 10, 'value': 0.1, 'step': 0.05, 'label': 'Gamma'})
123
+ ui.add_component('se_n_neighbors', gr.Slider, {'minimum': 6, 'maximum': 1000, 'value': 30, 'step': 1, 'label': 'Number of neighbors'})
124
+
125
+ ui.add_component('nca_init', gr.Dropdown, {'choices': ['auto', 'pca', 'lda', 'identity', 'random'], 'value': 'auto', 'label': 'Initialization of the linear transformation', 'interactive': True})
126
+ ui.add_component('nca_max_iter', gr.Slider, {'minimum': 5, 'maximum': 100, 'value': 50, 'step': 1, 'label': 'Maximum number of iterations'})
127
+
128
+
129
+ #Bindings
130
+
131
+ ui.bind('tsne_early_exaggeration', 'algorithm', 't-SNE')
132
+ ui.bind('tsne_perplexity', 'algorithm', 't-SNE')
133
+ ui.bind('tsne_learning_rate_dd', 'algorithm', 't-SNE')
134
+ ui.bind('tsne_learning_rate_s', 'tsne_learning_rate_dd', 'float')
135
+ ui.bind('tsne_n_iter', 'algorithm', 't-SNE')
136
+
137
+ ui.bind('n_neighbors', 'algorithm', 'Isomap')
138
+ ui.bind('isomap_p', 'algorithm', 'Isomap')
139
+
140
+ ui.bind('n_neighbors', 'algorithm', 'Locally Linear Embedding')
141
+ ui.bind('lle_method', 'algorithm', 'Locally Linear Embedding')
142
+
143
+ ui.bind('rte_n_estimators', 'algorithm', 'Random Trees Embedding')
144
+ ui.bind('rte_max_depth', 'algorithm', 'Random Trees Embedding')
145
+
146
+ ui.bind('se_affinity', 'algorithm', 'Spectral Embedding')
147
+ ui.bind('se_n_neighbors', 'se_affinity', 'nearest_neighbors')
148
+ ui.bind('se_gamma', 'se_affinity', 'rbf')
149
+
150
+ ui.bind('nca_init', 'algorithm', 'Neighborhood Component Analysis')
151
+ ui.bind('nca_max_iter', 'algorithm', 'Neighborhood Component Analysis')
152
+
153
+ btn = gr.Button('Submit')
154
+
155
+ with gr.Column():
156
+ plot = gr.Plot(label='')
157
+
158
+ def create_plot(*args):
159
+ kwargs = ui.get_kwargs(args)
160
+
161
+ algorithm = kwargs['algorithm']
162
+ n_components = int(kwargs['n_components'][0])
163
+
164
+ model = None
165
+ data = X
166
+
167
+ if algorithm == 'Linear Discriminant Analysis':
168
+ data = X.copy()
169
+ data.flat[:: X.shape[1] + 1] += 0.01 # Make X invertible
170
+ model = LinearDiscriminantAnalysis()
171
+
172
+ elif algorithm == 't-SNE':
173
+ model = TSNE(n_jobs=-1)
174
+ model.perplexity = kwargs['tsne_perplexity']
175
+ model.early_exaggeration = kwargs['tsne_early_exaggeration']
176
+ model.n_iter = kwargs['tsne_n_iter']
177
+ model.learning_rate = 'auto' if (kwargs['tsne_learning_rate_dd'] == 'auto') else kwargs['tsne_learning_rate_s']
178
+
179
+ elif algorithm == 'Truncated SVD':
180
+ model = TruncatedSVD()
181
+
182
+ elif algorithm == 'Isomap':
183
+ model = Isomap(n_jobs=-1)
184
+ model.n_neighbors = kwargs['n_neighbors']
185
+ model.p = kwargs['isomap_p']
186
+
187
+ elif algorithm == 'Locally Linear Embedding':
188
+ model = LocallyLinearEmbedding(n_jobs=-1)
189
+ model.n_neighbors = kwargs['n_neighbors']
190
+ model.method = kwargs['lle_method']
191
+
192
+ elif algorithm == 'Random Trees Embedding':
193
+ model = make_pipeline(RandomTreesEmbedding(n_estimators=kwargs['rte_n_estimators'], max_depth=kwargs['rte_max_depth']), TruncatedSVD(n_components=n_components))
194
+
195
+ elif algorithm == 'Spectral Embedding':
196
+ model = SpectralEmbedding(n_jobs=-1)
197
+ model.affinity = kwargs['se_affinity']
198
+ model.gamma = kwargs['se_gamma']
199
+ model.n_neighbors = kwargs['se_n_neighbors']
200
+
201
+ elif algorithm == 'Neighborhood Component Analysis':
202
+ model = NeighborhoodComponentsAnalysis()
203
+ model.init = kwargs['nca_init']
204
+ model.max_iter = kwargs['nca_max_iter']
205
+
206
+ # ===================
207
+ if algorithm != 'Random Trees Embedding':
208
+ model.n_components = n_components
209
+
210
+ projections = model.fit_transform(data, y)
211
+ projections = MinMaxScaler().fit_transform(projections)
212
+
213
+ fig = go.Figure()
214
+
215
+ for digit in digits.target_names:
216
+ subset = projections[y==digit]
217
+ rgba = plt.cm.tab10(digit/10)
218
+ color = f'rgba({rgba[0]}, {rgba[1]}, {rgba[2]}, 0.8)'
219
+ if n_components == 2:
220
+ fig.add_trace(go.Scatter(x=subset[:,0], y=subset[:,1], mode='text', text=str(digit), textfont={'size': 16, 'color': color}))
221
+ elif n_components == 3:
222
+ fig.add_trace(go.Scatter3d(x=subset[:,0], y=subset[:,1], z=subset[:,2], mode='text', text=str(digit), textfont={'size': 16, 'color': color}))
223
+
224
+ fig.update_traces(showlegend=False)
225
+
226
+ return fig
227
+
228
+ btn.click(create_plot, ui.all_components(), plot)
229
+ demo.load(create_plot, ui.all_components(), plot)
230
+
231
+ demo.launch(share=True)
232
+ # ====================================================================