synthcity.plugins.core.models.survival_analysis._base module
- class SurvivalAnalysisPlugin
Bases:
synthcity.plugins.core.serializable.Serializable
- abstract fit(X: pandas.core.frame.DataFrame, T: pandas.core.series.Series, Y: pandas.core.series.Series) Any
Training logic
- abstract static hyperparameter_space(**kwargs: Any) List[synthcity.plugins.core.distribution.Distribution]
Returns the hyperparameter space for the derived plugin.
- static load(buff: bytes) Any
- static load_dict(representation: dict) Any
- abstract static name() str
The name of the plugin.
- abstract predict(X: pandas.core.frame.DataFrame, time_horizons: List) pandas.core.frame.DataFrame
Predict risk
- classmethod sample_hyperparameters(*args: Any, **kwargs: Any) Dict[str, Any]
Sample value from the hyperparameter space for the current plugin.
- save() bytes
- save_dict() dict
- save_to_file(path: pathlib.Path) bytes
- static version() str
API version