9 Commits

Author SHA1 Message Date
76d3fed76f Enhance compute_disease_capture_at_k_fast to support return counts and update LandmarkEvaluator for backward compatibility with new metrics structure 2026-01-19 00:01:21 +08:00
d13fa430b7 Validate bin edges for Piecewise Exponential CIF Loss to ensure at least two finite edges and correct starting point 2026-01-18 20:54:47 +08:00
6de2132e84 Add fast vectorized implementation for Disease-Capture@K and enhance LandmarkEvaluator with profiling and correctness check options 2026-01-18 18:14:45 +08:00
6e76d67a10 Enhance _maybe_torch_compile and add _maybe_cudagraph_mark_step_begin for improved CUDA Graphs handling 2026-01-18 18:04:54 +08:00
a4b19b6e08 Enhance LandmarkEvaluator with model compilation and optimization options 2026-01-18 17:56:59 +08:00
014393a33f Add support for evaluation indices in LandmarkEvaluator class 2026-01-18 17:43:47 +08:00
0057bc0dd9 Refactor evaluation scripts for multi-GPU execution
- Removed `run_evaluations_multi_gpu.sh` script as it was redundant.
- Updated `run_experiments_multi_gpu.sh` to handle evaluation jobs instead of training.
- Changed command-line options to support evaluation-specific parameters.
- Implemented run directory discovery and validation for evaluation jobs.
- Enhanced logging to capture evaluation details and outputs.
- Added options for centralized output management and skipping existing results.
2026-01-18 17:38:20 +08:00
90dffc3211 Add evaluation scripts for age-bin time-dependent metrics and remove obsolete evaluation_time_dependent.py 2026-01-16 16:13:31 +08:00
34d8d8ce9d Add evaluation and utility functions for time-dependent metrics
- Introduced `evaluate.py` for time-dependent evaluation of models, including data loading and model inference.
- Added `evaluation_time_dependent.py` to compute various evaluation metrics such as AUC, average precision, and precision/recall at specified thresholds.
- Implemented CIF calculation methods in `losses.py` for different loss types, including exponential and piecewise exponential models.
- Created utility functions in `utils.py` for context selection and multi-hot encoding of events within specified horizons.
2026-01-16 14:55:09 +08:00