Add Weibull shape export scripts
This commit is contained in:
76
run_weibull_shape_exports.sh
Executable file
76
run_weibull_shape_exports.sh
Executable file
@@ -0,0 +1,76 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
# Export Weibull shape-parameter summaries for the all_future models trained
|
||||
# with smoking/alcohol/BMI extra information.
|
||||
#
|
||||
# Bash 4.2 compatible. Run from the DeepHealth repository root on the Linux
|
||||
# server, for example:
|
||||
#
|
||||
# bash run_weibull_shape_exports.sh
|
||||
#
|
||||
# Optional overrides:
|
||||
# PYTHON=python3 DEVICE=cuda BATCH_SIZE=128 NUM_WORKERS=4 ROW_BATCH_SIZE=512 \
|
||||
# bash run_weibull_shape_exports.sh
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
PYTHON="${PYTHON:-python}"
|
||||
DEVICE="${DEVICE:-cuda}"
|
||||
BATCH_SIZE="${BATCH_SIZE:-128}"
|
||||
NUM_WORKERS="${NUM_WORKERS:-4}"
|
||||
ROW_BATCH_SIZE="${ROW_BATCH_SIZE:-512}"
|
||||
LANDMARK_START="${LANDMARK_START:-40}"
|
||||
LANDMARK_STOP="${LANDMARK_STOP:-80}"
|
||||
LANDMARK_STEP="${LANDMARK_STEP:-5}"
|
||||
HORIZONS="${HORIZONS:-1,5,10}"
|
||||
|
||||
RUNS=(
|
||||
"runs/relative_weibull_all_future_pure_disease_20260620_095229"
|
||||
"runs/relative_mixed_all_future_pure_disease_20260620_132415"
|
||||
"runs/absolute_weibull_all_future_pure_disease_20260620_114816"
|
||||
"runs/absolute_mixed_all_future_pure_disease_20260620_161804"
|
||||
)
|
||||
|
||||
echo "Python: ${PYTHON}"
|
||||
echo "Device: ${DEVICE}"
|
||||
echo "Batch size: ${BATCH_SIZE}"
|
||||
echo "Workers: ${NUM_WORKERS}"
|
||||
echo "Row batch size: ${ROW_BATCH_SIZE}"
|
||||
echo "Horizons: ${HORIZONS}"
|
||||
echo
|
||||
|
||||
for run_path in "${RUNS[@]}"; do
|
||||
if [[ ! -d "${run_path}" ]]; then
|
||||
echo "[ERROR] Missing run directory: ${run_path}" >&2
|
||||
exit 1
|
||||
fi
|
||||
if [[ ! -f "${run_path}/best_model.pt" ]]; then
|
||||
echo "[ERROR] Missing checkpoint: ${run_path}/best_model.pt" >&2
|
||||
exit 1
|
||||
fi
|
||||
if [[ ! -f "${run_path}/train_config.json" ]]; then
|
||||
echo "[ERROR] Missing config: ${run_path}/train_config.json" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
output_path="${run_path}/weibull_shape_parameter_stats_test"
|
||||
echo "=== Exporting Weibull shape stats: ${run_path} ==="
|
||||
"${PYTHON}" export_weibull_shape_parameter_stats.py \
|
||||
--run_path "${run_path}" \
|
||||
--output_path "${output_path}" \
|
||||
--eval_split test \
|
||||
--device "${DEVICE}" \
|
||||
--batch_size "${BATCH_SIZE}" \
|
||||
--num_workers "${NUM_WORKERS}" \
|
||||
--row_batch_size "${ROW_BATCH_SIZE}" \
|
||||
--hidden_cache_dtype float32 \
|
||||
--landmark_start "${LANDMARK_START}" \
|
||||
--landmark_stop "${LANDMARK_STOP}" \
|
||||
--landmark_step "${LANDMARK_STEP}" \
|
||||
--horizons "${HORIZONS}" \
|
||||
--include_all_token_rho_summary
|
||||
echo "Wrote: ${output_path}"
|
||||
echo
|
||||
done
|
||||
|
||||
echo "All Weibull shape exports completed."
|
||||
Reference in New Issue
Block a user