skills/scikit-survival/scripts/train_survival_model.py

• resolve_schema · function · L34-L79 — def resolve_schema( frame: Any, *, event_column: str, time_column: str, numeric_columns: list[str], categorical_columns: list[str], ) -> tuple[Any, Any, dict[str, Any]]
• build_pipeline · function · L82-L174 — def build_pipeline( model_name: str, numeric_columns: list[str], categorical_columns: list[str], *, seed: int, )
• parameter_grid · function · L177-L191 — def parameter_grid(model_name: str) -> dict[str, list[Any]]
• _event_field · function · L194-L195 — def _event_field(y: Any) -> str
• _time_field · function · L198-L199 — def _time_field(y: Any) -> str
• stratified_splits · function · L202-L217 — def stratified_splits(y: Any, folds: int, *, seed: int) -> list[tuple[Any, Any]]
• uno_concordance · function · L220-L245 — def uno_concordance(y_train: Any, y_test: Any, risk: Any) -> dict[str, Any]
• nested_tune · function · L248-L324 — def nested_tune( X: Any, y: Any, *, model_name: str, numeric_columns: list[str], categorical_columns: list[str], outer_folds: int, inner_folds: int, seed: int, ) -> tuple[list[dict[str, Any]], dict[str, Any]]
• prediction_archive · function · L327-L364 — def prediction_archive( estimator: Any, X_test: Any, y_train: Any, y_test: Any, risk: Any, ) -> dict[str, Any]
• train_and_report · function · L367-L490 — def train_and_report( frame: Any, *, event_column: str, time_column: str, numeric_columns: list[str], categorical_columns: list[str], model_name: str, test_fraction: float, seed: int, tune: bool, outer_folds: int, inner_folds: int, ) -> tuple[Any, Any, Any, dict[str, Any]]
• build_parser · function · L493-L528 — def build_parser() -> argparse.ArgumentParser
• main · function · L531-L579 — def main(argv: list[str] | None = None) -> int