{
    "activation": "ReLU",
    "adding_bond_types": true,
    "adding_h": false,
    "aggregation": "mean",
    "aggregation_norm": 100,
    "atom_constraints": [],
    "atom_descriptor_scaling": true,
    "atom_descriptors": null,
    "atom_descriptors_path": null,
    "atom_descriptors_size": 0,
    "atom_features_size": 0,
    "atom_messages": false,
    "atom_targets": [],
    "batch_size": 50,
    "bias": false,
    "bias_solvent": false,
    "bond_constraints": [],
    "bond_descriptor_scaling": true,
    "bond_descriptors": null,
    "bond_descriptors_path": null,
    "bond_descriptors_size": 0,
    "bond_features_size": 0,
    "bond_targets": [],
    "cache_cutoff": 10000,
    "checkpoint_dir": null,
    "checkpoint_frzn": null,
    "checkpoint_path": null,
    "checkpoint_paths": null,
    "class_balance": false,
    "config_path": "../data/args_files/optimized_configs.json",
    "constraints_path": null,
    "crossval_index_dir": null,
    "crossval_index_file": null,
    "crossval_index_sets": null,
    "cuda": true,
    "data_path": "../data/crossval_splits/all_amine_split_for_paper/cv_4/train.csv",
    "data_weights_path": "../data/crossval_splits/all_amine_split_for_paper/cv_4/train_weights.csv",
    "dataset_type": "regression",
    "depth": 4,
    "depth_solvent": 3,
    "device": {
        "_string": "cuda",
        "_type": "python_object (type = device)",
        "_value": "gASVHwAAAAAAAACMBXRvcmNolIwGZGV2aWNllJOUjARjdWRhlIWUUpQu"
    },
    "dropout": 0.1,
    "empty_cache": false,
    "ensemble_size": 1,
    "epochs": 50,
    "evidential_regularization": 0,
    "explicit_h": false,
    "extra_metrics": [],
    "features_generator": null,
    "features_only": false,
    "features_path": [
        "../data/crossval_splits/all_amine_split_for_paper/cv_4/train_extra_x.csv"
    ],
    "features_scaling": true,
    "features_size": null,
    "ffn_hidden_size": 600,
    "ffn_num_layers": 3,
    "final_lr": 0.0001,
    "folds_file": null,
    "freeze_first_only": false,
    "frzn_ffn_layers": 0,
    "gpu": null,
    "grad_clip": null,
    "hidden_size": 600,
    "hidden_size_solvent": 300,
    "ignore_columns": null,
    "ignore_nan_metrics": false,
    "init_lr": 0.0001,
    "is_atom_bond_targets": false,
    "keeping_atom_map": false,
    "log_frequency": 10,
    "loss_function": "mse",
    "max_data_size": null,
    "max_lr": 0.001,
    "metric": "rmse",
    "metrics": [
        "rmse"
    ],
    "minimize_score": true,
    "mpn_shared": false,
    "multiclass_num_classes": 3,
    "no_adding_bond_types": false,
    "no_atom_descriptor_scaling": false,
    "no_bond_descriptor_scaling": false,
    "no_cache_mol": false,
    "no_cuda": false,
    "no_features_scaling": false,
    "no_shared_atom_bond_ffn": false,
    "num_folds": 1,
    "num_lrs": 1,
    "num_tasks": 1,
    "num_workers": 8,
    "number_of_molecules": 1,
    "overwrite_default_atom_features": false,
    "overwrite_default_bond_features": false,
    "phase_features_path": null,
    "pytorch_seed": 0,
    "quantile_loss_alpha": 0.1,
    "quantiles": [],
    "quiet": false,
    "reaction": false,
    "reaction_mode": "reac_diff",
    "reaction_solvent": false,
    "reproducibility": {
        "command_line": "python main_script.py train all_amine_split_for_paper",
        "git_has_uncommitted_changes": true,
        "git_root": "/media/andersonxps/wd_4tb/evan/LNP_ML",
        "git_url": "https://github.com/jswitten/LNP_ML/tree/167822980dc26ba65c5c14539c4ce12b81b0b8f3",
        "time": "Tue Jul 30 10:40:44 2024"
    },
    "resume_experiment": false,
    "save_dir": "../data/crossval_splits/all_amine_split_for_paper/cv_4",
    "save_preds": false,
    "save_smiles_splits": false,
    "seed": 42,
    "separate_test_atom_descriptors_path": null,
    "separate_test_bond_descriptors_path": null,
    "separate_test_constraints_path": null,
    "separate_test_features_path": [
        "../data/crossval_splits/all_amine_split_for_paper/cv_4/test_extra_x.csv"
    ],
    "separate_test_path": "../data/crossval_splits/all_amine_split_for_paper/cv_4/test.csv",
    "separate_test_phase_features_path": null,
    "separate_val_atom_descriptors_path": null,
    "separate_val_bond_descriptors_path": null,
    "separate_val_constraints_path": null,
    "separate_val_features_path": [
        "../data/crossval_splits/all_amine_split_for_paper/cv_4/valid_extra_x.csv"
    ],
    "separate_val_path": "../data/crossval_splits/all_amine_split_for_paper/cv_4/valid.csv",
    "separate_val_phase_features_path": null,
    "shared_atom_bond_ffn": true,
    "show_individual_scores": false,
    "smiles_columns": [
        "smiles"
    ],
    "spectra_activation": "exp",
    "spectra_phase_mask_path": null,
    "spectra_target_floor": 1e-08,
    "split_key_molecule": 0,
    "split_sizes": [
        1.0,
        0.0,
        0.0
    ],
    "split_type": "random",
    "target_columns": null,
    "target_weights": null,
    "task_names": [
        "quantified_delivery"
    ],
    "test": false,
    "test_fold_index": null,
    "train_data_size": null,
    "undirected": false,
    "use_input_features": true,
    "val_fold_index": null,
    "warmup_epochs": 2.0,
    "weights_ffn_num_layers": 2
}