|
31 | 31 | {'name':'train_bool', 'type':candle.str2bool,'default':True,'help':'Invoke training'},
|
32 | 32 | {'name':'eval_bool', 'type':candle.str2bool,'default':False,'help':'Use model for inference'},
|
33 | 33 | {'name':'home_dir','help':'Home Directory','type':str,'default':'.'},
|
34 |
| -{'name':'config_file','help':'Config File','type':str,'default':os.path.join(file_path, 'p2b1_default_model.txt')}, |
| 34 | +#{'name':'config_file','help':'Config File','type':str,'default':os.path.join(file_path, 'p2b1_default_model.txt')}, |
35 | 35 | {'name':'weight_path','help':'Trained Model Pickle File','type':str,'default':None},
|
36 | 36 | {'name':'base_memo','help':'Memo','type':str,'default':None},
|
37 |
| -{'name':'seed', 'type':candle.str2bool,'default':False,'help':'Random Seed'}, |
| 37 | +#{'name':'seed_bool', 'type':candle.str2bool,'default':False,'help':'Random Seed'}, |
38 | 38 | {'name':'case','help':'[Full, Center, CenterZ]','type':str,'default':'Full'},
|
39 | 39 | {'name':'fig_bool', 'type':candle.str2bool,'default':False,'help':'Generate Prediction Figure'},
|
40 | 40 | {'name':'set_sel','help':'[3k_Disordered, 3k_Ordered, 3k_Ordered_and_gel, 6k_Disordered, 6k_Ordered, 6k_Ordered_and_gel]','type':str,'default':'3k_Disordered'},
|
41 | 41 | {'name':'conv_bool', 'type':candle.str2bool, 'default':True, 'help':'Invoke training using 1D Convs for inner AE'},
|
42 | 42 | {'name':'full_conv_bool', 'type':candle.str2bool, 'default':False, 'help':'Invoke training using fully convolutional NN for inner AE'},
|
43 | 43 | {'name':'type_bool', 'type':candle.str2bool, 'default':True, 'help':'Include molecule type information in desining AE'},
|
44 | 44 | {'name':'nbr_type', 'type':str, 'default':'relative', 'help':'Defines the type of neighborhood data to use. [relative, invariant]'},
|
45 |
| -{'name':'backend', 'help':'Keras Backend', 'type':str, 'default':'tensorflow'} |
| 45 | +{'name':'backend', 'help':'Keras Backend', 'type':str, 'default':'tensorflow'}, |
| 46 | +{'name':'cool', 'help':'Boolean: cool learning rate', 'type':candle.str2bool, 'default':False}, |
| 47 | +{'name':'data_set', 'help':'Data set for training', 'type':str, 'default':None}, |
| 48 | +{'name':'l2_reg', 'help':'Regularization parameter', 'type':float, 'default':None}, |
| 49 | +{'name':'molecular_nbrs', 'help':'Data dimension for molecular autoencoder', 'type':int, 'default':None}, |
| 50 | +{'name':'molecular_nonlinearity', 'help':'Activation for molecular netowrk', 'type':str, 'default':None}, |
| 51 | +{'name':'molecular_num_hidden', 'nargs':'+', 'help':'Layer sizes for molecular network', 'type':int, 'default':None}, |
| 52 | +{'name':'noise_factor', 'help':'Noise factor', 'type':float, 'default':None}, |
| 53 | +{'name':'num_hidden', 'nargs':'+', 'help':'Dense layer specification', 'type':int, 'default':None}, |
| 54 | +{'name':'sampling_density', 'help':'Sampling density', 'type':float, 'default':None} |
46 | 55 | ]
|
47 | 56 |
|
48 | 57 | required = [
|
|
61 | 70 | 'molecular_num_hidden',
|
62 | 71 | 'molecular_nonlinearity',
|
63 | 72 | 'molecular_nbrs',
|
64 |
| - 'drop_prob', |
| 73 | + 'dropout', |
65 | 74 | 'l2_reg',
|
66 | 75 | 'sampling_density',
|
67 | 76 | 'save_path'
|
|
0 commit comments