Remove legacy event and mixed distribution paths
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user