Remove legacy event and mixed distribution paths

This commit is contained in:
2026-08-01 14:23:18 +08:00
parent dfb22adf2d
commit de6f9b75b9
22 changed files with 370 additions and 463 deletions

View File

@@ -207,19 +207,15 @@ class IPCWCalibrationMetricTests(unittest.TestCase):
rho = np.asarray([0.8, 1.0, 1.2, 1.5], dtype=np.float32)
horizons = np.asarray([0.1, 1.0, 5.0], dtype=np.float32)
for dist_mode, token, death_idx, selected_rho in (
("exponential", 4, 9, None),
("weibull", 4, 9, rho),
("mixed", 9, 9, rho),
("mixed", 4, 9, None),
for dist_mode, selected_rho in (
("exponential", None),
("weibull", rho),
):
actual = _risk_probability_matrix(
logits=logits,
rho=selected_rho,
horizons=horizons,
dist_mode=dist_mode,
token=token,
death_idx=death_idx,
)
expected = np.vstack(
[
@@ -229,8 +225,6 @@ class IPCWCalibrationMetricTests(unittest.TestCase):
score_mode="risk",
horizon=float(horizon),
dist_mode=dist_mode,
token=token,
death_idx=death_idx,
)
for horizon in horizons
]
@@ -271,7 +265,6 @@ class IPCWCalibrationMetricTests(unittest.TestCase):
"label_id_to_code": {4: "D4", 5: "D5"},
"dist_mode": "exponential",
"horizons": np.asarray([1.0, 5.0], dtype=np.float32),
"death_index": 9,
"min_cases": 1,
"min_controls": 1,
"max_ipcw_weight": 0.0,