-
Notifications
You must be signed in to change notification settings - Fork 4
/
note.txt
57 lines (43 loc) · 1.32 KB
/
note.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# NAS-related
main.py -- integrating training/test/
utils.py -- data preprocessing
trainer.py -- integrating NAS functions: maintaining the supernet, the controller and the search process
network.py -- the supernet class
dataset.py -- ezgeno training set testing set preparing
controller.py -- the controller class
main.py <---- utils.py
|-- trainer.py <----- network.py
|-- controller.py
# Gradcam-related
visualize.py -- show gradcam
visulaize.py <--- X
## utils.py
* function get_variable: unknown
* function keydefaultdict: unknown
* function get_mean_and_std: unknown
* function init_params: unknown
* function onehot_encode_sequences: pass
* function choose_optimizer: move into trainer
## dataset.py
* class trainset: aggregate tf/epi
* class testset: aggregate tf/eps
* function prepare_all_data: aggregate tf/epi
## network.py
* class ezGenoModel: aggregate tf/epi, self-defined filter_size
## controller.py
* class contorller: aggregate tf/epi
## trainer.py
* class ezGenoTrainer
** function train_supernet
** function train_controller
** function train_subnet
** function test
** function train
** function load_model
** function save_model
## visualize.py
* class FeatureExtractor
* class ModelOutputs
* class GradCam
* class gradCamTestset
* function show_grad_cam