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