{
    "exp_info": {
        "main_folder": "exp_output",
        "exp_name": "overall_result",
        "annotation": "HCC_MedFuse"
    },
    "dataset": {
        "module": "datasets.ehr_dataset",
        "type": "GetDataset",
        "data_path": "./data/HCC",
        "BasicStatisticInfo": {
            "drop_outlier": true,
            "outlier_fence": 3.0,
            "statistic_plot": false,
            "plot_output_dir": "TODO"
        },
        "train": {
            "GetDataset": {
                "mode": "train",
                "do_kfold": false,
                "k": null,
                "do_augmentation": false,
                "do_normalization": true
            }
        },
        "test": {
            "GetDataset": {
                "mode": "test",
                "do_kfold": false,
                "k": null,
                "do_augmentation": false,
                "do_normalization": true
            }
        }
    },
    "dataloader": {
        "train": {
            "batch_size": 256,
            "shuffle": true,
            "num_workers": 4
        },
        "valid": {
            "batch_size": 256,
            "shuffle": false,
            "num_workers": 4
        },
        "test": {
            "batch_size": 256,
            "shuffle": false,
            "num_workers": 4
        }
    },
    "model": {
        "module": "models.medfuse",
        "type": "medfuse",
        "kwargs": {
            "d_model": 144,
            "num_heads": 1,
            "ff_dim": 144,
            "ff_dropout": 0.5,
            "attn_dropout": 0.05,
            "norm_type": "LayerNorm",
            "only_mask_first": false,
            "seq_len": 4,
            "NUM_LEN": 30,
            "CAT_LEN": 8,
            "num_layers": 8,
            "decoder_down_factor": 2,
            "decoder_dropout": 0.45,
            "output_size": 1,
            "num_embedding_module_kwargs": {
                "independent_padding": true,
                "max_norm": null,
                "value_proj_dim": 32,
                "k_gates": 144
            },
            "cat_embedding_module_kwargs": {
                "independent_padding": true,
                "max_norm": null
            }
        }
    },
    "loss": {
        "module": "loss.focal_loss",
        "type": "FocalLoss",
        "kwargs": {
            "alpha": 0.5,
            "gamma": 2.0,
            "reduction": "mean",
            "eps": 0
        }
    },
    "optimizer": {
        "module": "torch.optim",
        "type": "Adam",
        "kwargs": {
            "lr": 3.00009e-05
        }
    },
    "lr_scheduler": {
        "module": null,
        "type": null
    },
    "trainer": {
        "module": "trainer.base_trainer_HCC",
        "type": "BaseTrainer",
        "kwargs": {
            "do_early_stop": false,
            "checkpoint_period": 5,
            "plot_process": true,
            "resume": false,
            "resume_checkpoint_path": "TODO"
        },
        "train_kwargs": {
            "epoch": 150
        }
    },
    "tester": {
        "module": "tester.base_tester_HCC",
        "type": "BaseTester",
        "kwargs": {
            "plot_probability_distribution": true
        }
    }
}