| { | |
| "best_val_loss": 0.01979512721300125, | |
| "comet_ai_key": null, | |
| "context_observations": { | |
| "add_rem": true, | |
| "empirical_number_of_obs": 2, | |
| "max_num_obs": 15, | |
| "max_past": 5, | |
| "min_past": 3, | |
| "past_time_ratio": 0.1, | |
| "split_past_future": false, | |
| "type": "pk_peak_half_life" | |
| }, | |
| "debug_test": false, | |
| "dosing": { | |
| "logdose_mean_range": [ | |
| -2.0, | |
| 2.0 | |
| ], | |
| "logdose_std_range": [ | |
| 0.1, | |
| 0.5 | |
| ], | |
| "num_individuals": 10, | |
| "route_options": [ | |
| "oral", | |
| "iv" | |
| ], | |
| "route_weights": [ | |
| 0.8, | |
| 0.2 | |
| ], | |
| "same_route": true, | |
| "time": 0.0 | |
| }, | |
| "experiment_dir": "/work/ojedamarin/Projects/Pharma/Results/comet/aistats/c2b9bc645f434a6dbac001e145eae4e4", | |
| "experiment_indentifier": null, | |
| "experiment_name": "aistats", | |
| "hf_model_card_path": [ | |
| "hf_model_cards", | |
| "AICME-PK_Readme.md" | |
| ], | |
| "hf_model_name": "AICMEPK_cluster", | |
| "hugging_face_token": null, | |
| "meta_study": { | |
| "V_tmag_range": [ | |
| 0.001, | |
| 0.001 | |
| ], | |
| "V_tscl_range": [ | |
| 1, | |
| 5 | |
| ], | |
| "drug_id_options": [ | |
| "Drug_A", | |
| "Drug_B", | |
| "Drug_C" | |
| ], | |
| "k_1p_tmag_range": [ | |
| 0.01, | |
| 0.02 | |
| ], | |
| "k_1p_tscl_range": [ | |
| 1, | |
| 5 | |
| ], | |
| "k_a_tmag_range": [ | |
| 0.01, | |
| 0.02 | |
| ], | |
| "k_a_tscl_range": [ | |
| 1, | |
| 5 | |
| ], | |
| "k_e_tmag_range": [ | |
| 0.01, | |
| 0.02 | |
| ], | |
| "k_e_tscl_range": [ | |
| 1, | |
| 5 | |
| ], | |
| "k_p1_tmag_range": [ | |
| 0.01, | |
| 0.02 | |
| ], | |
| "k_p1_tscl_range": [ | |
| 1, | |
| 5 | |
| ], | |
| "log_V_mean_range": [ | |
| 2, | |
| 8 | |
| ], | |
| "log_V_std_range": [ | |
| 0.2, | |
| 0.6 | |
| ], | |
| "log_k_1p_mean_range": [ | |
| -4, | |
| 0 | |
| ], | |
| "log_k_1p_std_range": [ | |
| 0.2, | |
| 0.6 | |
| ], | |
| "log_k_a_mean_range": [ | |
| -1, | |
| 2 | |
| ], | |
| "log_k_a_std_range": [ | |
| 0.2, | |
| 0.6 | |
| ], | |
| "log_k_e_mean_range": [ | |
| -5, | |
| 0 | |
| ], | |
| "log_k_e_std_range": [ | |
| 0.2, | |
| 0.6 | |
| ], | |
| "log_k_p1_mean_range": [ | |
| -4, | |
| -1 | |
| ], | |
| "log_k_p1_std_range": [ | |
| 0.2, | |
| 0.6 | |
| ], | |
| "num_individuals_range": [ | |
| 5, | |
| 10 | |
| ], | |
| "num_peripherals_range": [ | |
| 1, | |
| 3 | |
| ], | |
| "rel_ruv_range": [ | |
| 0.001, | |
| 0.01 | |
| ], | |
| "solver_method": "rk4", | |
| "time_num_steps": 100, | |
| "time_start": 0.0, | |
| "time_stop": 16.0 | |
| }, | |
| "mix_data": { | |
| "evaluate_prediction_steps_past": 5, | |
| "keep_tempfile": true, | |
| "log_transform": false, | |
| "n_of_databatches": null, | |
| "n_of_permutations": 3, | |
| "n_of_target_individuals": 1, | |
| "normalize_by_max": true, | |
| "normalize_time": true, | |
| "recreate_tempfile": false, | |
| "store_in_tempfile": true, | |
| "tempfile_path": [ | |
| "preprocessed", | |
| "simulated_ou_as_rates" | |
| ], | |
| "test_empirical_datasets": [ | |
| "cesarali/lenuzza-2016", | |
| "cesarali/Theophylline" | |
| ], | |
| "test_size": 64, | |
| "tqdm_progress": false, | |
| "train_size": 128, | |
| "val_size": 64, | |
| "z_score_normalization": false | |
| }, | |
| "model_type": "node_pk", | |
| "my_results_path": "/work/ojedamarin/Projects/Pharma/Results/", | |
| "name_str": "AICMEPK", | |
| "network": { | |
| "activation": "ReLU", | |
| "aggregator_num_heads": 8, | |
| "aggregator_type": "attention", | |
| "combine_latent_mode": "mlp", | |
| "cov_proj_dim": 16, | |
| "decoder_attention_layers": 2, | |
| "decoder_hidden_dim": 512, | |
| "decoder_name": "TransformerDecoder", | |
| "decoder_num_layers": 4, | |
| "decoder_rnn_hidden_dim": 256, | |
| "drift_activation": "Tanh", | |
| "drift_num_layers": 2, | |
| "dropout": 0.1, | |
| "encoder_rnn_hidden_dim": 256, | |
| "exclusive_node_step": false, | |
| "ignore_logvar": true, | |
| "individual_encoder_name": "RNNContextEncoder", | |
| "individual_encoder_number_of_heads": 4, | |
| "init_hidden_num_layers": 4, | |
| "input_encoding_hidden_dim": 128, | |
| "kl_weight": 1.0, | |
| "loss_name": "log_nll", | |
| "node_step": true, | |
| "norm": "layer", | |
| "output_head_num_layers": 3, | |
| "prediction_latent_deterministic": false, | |
| "prediction_only": false, | |
| "reconstruction_only": false, | |
| "rnn_decoder_number_of_layers": 4, | |
| "rnn_individual_encoder_number_of_layers": 4, | |
| "study_latent_deterministic": false, | |
| "time_obs_encoder_hidden_dim": 256, | |
| "time_obs_encoder_output_dim": 256, | |
| "use_attention": true, | |
| "use_invariance_loss": true, | |
| "use_kl_i": true, | |
| "use_kl_i_np": true, | |
| "use_kl_init": true, | |
| "use_kl_s": true, | |
| "use_self_attention": true, | |
| "use_time_deltas": true, | |
| "zi_latent_dim": 256 | |
| }, | |
| "run_index": 0, | |
| "tags": [ | |
| "AISTATS-2026", | |
| "AICME-T-PK" | |
| ], | |
| "target_observations": { | |
| "add_rem": true, | |
| "empirical_number_of_obs": 2, | |
| "max_num_obs": 15, | |
| "max_past": 5, | |
| "min_past": 3, | |
| "past_time_ratio": 0.1, | |
| "split_past_future": true, | |
| "type": "pk_peak_half_life" | |
| }, | |
| "tracking": { | |
| "best": { | |
| "val_rmse": { | |
| "epoch": null, | |
| "step": null, | |
| "value": 0.01979512721300125 | |
| } | |
| }, | |
| "meta": {} | |
| }, | |
| "train": { | |
| "amsgrad": false, | |
| "batch_size": 128, | |
| "betas": [ | |
| 0.9, | |
| 0.999 | |
| ], | |
| "empirical_every_pct": 0.1, | |
| "epochs": 10, | |
| "eps": 1e-08, | |
| "generative_image_ids": null, | |
| "generative_image_ids_empirical_during": null, | |
| "generative_image_ids_empirical_end": [ | |
| "new_individuals_images" | |
| ], | |
| "generative_image_ids_val": [ | |
| "new_individuals_images" | |
| ], | |
| "generative_metric_ids": null, | |
| "generative_metric_ids_empirical_during": null, | |
| "generative_metric_ids_empirical_end": [ | |
| "new_individuals_metrics" | |
| ], | |
| "generative_metric_ids_val": [], | |
| "gradient_clip_val": 0.5, | |
| "learning_rate": 0.0001, | |
| "log_generative_empirical_during": false, | |
| "log_generative_empirical_end": true, | |
| "log_generative_val": true, | |
| "log_images_empirical_during": false, | |
| "log_images_empirical_end": true, | |
| "log_images_val": true, | |
| "log_interval": 1, | |
| "log_predictive_empirical_during": false, | |
| "log_predictive_empirical_end": true, | |
| "log_predictive_val": true, | |
| "models_end_of_training": [ | |
| "last", | |
| "best" | |
| ], | |
| "num_batch_plot": 1, | |
| "num_workers": 8, | |
| "optimizer_name": "AdamW", | |
| "persistent_workers": true, | |
| "predictive_image_ids": null, | |
| "predictive_image_ids_empirical_during": null, | |
| "predictive_image_ids_empirical_end": [ | |
| "prediction_images" | |
| ], | |
| "predictive_image_ids_val": [ | |
| "prediction_images" | |
| ], | |
| "predictive_metric_ids": null, | |
| "predictive_metric_ids_empirical_during": null, | |
| "predictive_metric_ids_empirical_end": [ | |
| "prediction_metrics" | |
| ], | |
| "predictive_metric_ids_val": [], | |
| "scheduler_name": "CosineAnnealingLR", | |
| "scheduler_params": { | |
| "T_max": 1000, | |
| "eta_min": 5e-05, | |
| "last_epoch": -1 | |
| }, | |
| "shuffle_val": true, | |
| "val_every_pct": 0.1, | |
| "weight_decay": 0.0001 | |
| }, | |
| "transformers_version": "4.52.4", | |
| "upload_to_hf_hub": true, | |
| "verbose": false | |
| } | |