synthcity.plugins.core.models.survival_analysis.loader module

generate_dataset_for_horizon(X: pandas.core.frame.DataFrame, T: pandas.core.frame.DataFrame, Y: pandas.core.frame.DataFrame, horizon_days: int) → Tuple[pandas.core.frame.DataFrame, pandas.core.frame.DataFrame, pandas.core.frame.DataFrame]

Generate the dataset at a certain time horizon. Useful for classifiers.

Parameters

• X – pd.DataFrame, the feature set

• T – pd.DataFrame, days to event or censoring

• Y – pd.DataFrame, outcome or censoring

• horizon_days – int, days to the expected horizon

Returns

the feature set for that horizon T: days to event or censoring Y: Outcome or censoring

Return type

X

get_model_template(model: str) → Any