File size: 1,539 Bytes
30c86cf
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
65
66
67
68
69
70
{
    "drug_sensitivity_min_max": true,
    "gene_expression_min_max": false,
    "gene_expression_standardize": false,
    "augment_smiles": false,
    "canonical": false,
    "kekulize": false,
    "all_bonds_explicit": false,
    "all_hs_explicit": false,
    "randomize": false,
    "remove_bonddir": false,
    "remove_chirality": false,
    "selfies": false,
    "smiles_start_stop_token": true,
    "number_of_genes": 2128,
    "smiles_padding_length": 465,
    "stacked_dense_hidden_sizes": [
        512
    ],
    "activation_fn": "relu",
    "dropout": 0.4,
    "batch_norm": true,
    "filters": [
        64,
        64,
        64
    ],
    "multiheads": [
        4,
        4,
        4,
        4
    ],
    "smiles_embedding_size": 16,
    "kernel_sizes": [
        [
            3,
            16
        ],
        [
            5,
            16
        ],
        [
            11,
            16
        ]
    ],
    "smiles_attention_size": 64,
    "embed_scale_grad": false,
    "final_activation": true,
    "gene_to_dense": false,
    "batch_size": 2048,
    "dataset_device": "cuda",
    "lr": 0.001,
    "optimizer": "adam",
    "loss_fn": "mse",
    "epochs": 200,
    "save_model": 25,
    "smiles_vocabulary_size": 108,
    "drug_sensitivity_processing_parameters": {
        "processing": "min_max",
        "parameters": {
            "min": -11.998083341987641,
            "max": 12.359055999999999
        }
    },
    "gene_expression_processing_parameters": {},
    "number_of_parameters": 7217361
}