Add LogNormalBasisHazardLoss implementation and update training configuration

2026-01-13 15:59:20 +08:00
parent d8b322cbee
commit 1df02d85d7
5 changed files with 431 additions and 15 deletions
--- a/prepare_data.py
+++ b/prepare_data.py
@@ -4,7 +4,8 @@ import numpy as np                # Numerical operations

 # CSV mapping field IDs to human-readable names
 field_map_file = "field_ids_enriched.csv"
-field_dict = {}                             # Map original field ID -> new column name
+# Map original field ID -> new column name
+field_dict = {}
 tabular_fields = []                         # List of tabular feature column names
 with open(field_map_file, "r", encoding="utf-8") as f:  # Open the field mapping file
    next(f)  # skip header line