artifacts/model.json

{
    "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
}