HoTS/input_config.json at master · GIST-CSBL/HoTS · GitHub

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
{
    "dti_dir": "./SampleData/DTI/Training/Training_DTI_Sample.csv",
    "drug_dir": "./SampleData/DTI/Training/Training_Compound_Sample.csv",
    "protein_dir": "./SampleData/DTI/Training/Training_Protein_Sample.csv",
    "hots_dir": "./SampleData/HoTS/Training_HoTS.tsv",
    "validation_dti_dir": "./SampleData/DTI/Validation/Validation_DTI.csv",
    "validation_drug_dir": "./SampleData/DTI/Validation/Validation_Compound.csv",
    "validation_protein_dir": "./SampleData/DTI/Validation/Validation_Protein.csv",
    "validation_hots_dir": "./SampleData/HoTS/Validation_HoTS.tsv",
    "window_sizes": [10, 15, 20, 25, 30],
    "n_filters": 64,
    "drug_layers": [512, 128],
    "hots_dimension": 128,
    "n_heads": 4,
    "n_transformers_hots": 2,
    "n_transformers_dti": 4,
    "hots_fc_layers": [256, 64],
    "dti_fc_layers": [256, 64],
    "anchors": [9],
    "grid_size": 20,
    "learning_rate": 0.0001,
    "n_pretrain": 1,
    "n_epochs": 1,
    "hots_ratio": 1,
    "retina_loss": 2.0,
    "confidence_loss": 1,
    "regression_loss": 0.1,
    "negative_loss": 0.1,
    "drug_len": 2048,
    "radius": 2,
    "activation": "gelu",
    "dropout": 0.1,
    "batch_size": 16,
    "decay": 0.0001,
    "output": "./Saved_model"
}