-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetting_pharmacophore.yaml
64 lines (60 loc) · 1.53 KB
/
setting_pharmacophore.yaml
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
57
58
59
60
61
62
63
64
# Basic setting
c_val: 0.2
# threshold_type: [time, generation_num]
threshold_type: generation_num
#hours: 0.01
generation_num: 30000
output_dir: result/pharmacophore0609
model_setting:
model_json: model/user_trained/model.tf25.json
model_weight: model/user_trained/model.tf25.best.ckpt.h5
token: model/user_trained/tokens.pkl
# model_json: model/model_chembl220k.tf25.json
# model_weight: model/model_chembl220k.tf25.best.ckpt.h5
#token: model/tokens_chembl220k.pkl
reward_setting:
reward_module: reward.pharmacophore_reward
reward_class: Pharmacophore_reward
# Advanced setting
expansion_threshold: 0.995
simulation_num: 3
flush_threshold: -1
policy_setting:
policy_module: policy.ucb1
policy_class: Ucb1
# Restart setting
save_checkpoint: False
restart: False
checkpoint_file: chemtsv2.ckpt.pkl
# Filter setting
use_lipinski_filter: False
lipinski_filter:
module: filter.lipinski_filter
class: LipinskiFilter
type: rule_of_5
use_radical_filter: True
radical_filter:
module: filter.radical_filter
class: RadicalFilter
use_pubchem_filter: False
pubchem_filter:
module: filter.pubchem_filter
class: PubchemFilter
use_sascore_filter: True
sascore_filter:
module: filter.sascore_filter
class: SascoreFilter
threshold: 4.5
use_ring_size_filter: False
ring_size_filter:
module: filter.ring_size_filter
class: RingSizeFilter
threshold: 6
use_pains_filter: False
pains_filter:
module: filter.pains_filter
class: PainsFilter
type: [pains_a]
include_filter_result_in_reward: False
#leaf_parallel: True
#leaf_parallel_num: 20