synthcity.plugins.core.models.time_series_survival._base module

class TimeSeriesSurvivalPlugin

Bases: synthcity.plugins.core.serializable.Serializable

abstract fit(static: Optional[numpy.ndarray], temporal: numpy.ndarray, observation_times: numpy.ndarray, T: numpy.ndarray, Y: numpy.ndarray) → 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(static: Optional[numpy.ndarray], temporal: numpy.ndarray, observation_times: numpy.ndarray, time_horizons: List) → numpy.ndarray

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