discotime.utils.typing module

class discotime.utils.typing.LabelTransformer(*args, **kwargs)

Bases: Transformer, Protocol

property cuts: ndarray[Any, dtype[floating]]

fit(time: Iterable[int | int64 | float | float64], event: Iterable[int | int64]) → None

fit_transform(time: Iterable[int | int64 | float | float64], event: Iterable[int | int64]) → tuple[numpy.ndarray[Any, numpy.dtype[numpy.integer]], numpy.ndarray[Any, numpy.dtype[numpy.integer]]]

property max_time: int | int64 | float | float64

transform(time: Iterable[int | int64 | float | float64], event: Iterable[int | int64]) → tuple[numpy.ndarray[Any, numpy.dtype[numpy.integer]], numpy.ndarray[Any, numpy.dtype[numpy.integer]]]

class discotime.utils.typing.SurvData(features, event_time_disc, event_status_disc, event_time_cont, event_status_cont)

Bases: NamedTuple

event_status_cont: ndarray[Any, dtype[int64]]

Alias for field number 4

event_status_disc: ndarray[Any, dtype[int64]]

Alias for field number 2

event_time_cont: ndarray[Any, dtype[float64]]

Alias for field number 3

event_time_disc: ndarray[Any, dtype[int64]]

Alias for field number 1

features: ndarray[Any, dtype[float64]]

Alias for field number 0

class discotime.utils.typing.Transformer(*args, **kwargs)

Bases: Protocol

fit(*args, **kwargs) → None

fit_transform(*args, **kwargs) → Any

transform(*args, **kwargs) → Any