logml.survival_analysis.artifacts.base

Classes

BaseArtifact(sa_setup, global_params, ...)

Base class for survival analysis artifacts.

class logml.survival_analysis.artifacts.base.BaseArtifact(sa_setup: SurvivalAnalysisSetup, global_params: dict, output_structure: logml.survival_analysis.BaseSAMethodsOutputStructure)

Bases: object

Base class for survival analysis artifacts.

LABEL = None

build(container: logml.data.datasets.survival_dataset.UnivariateSurvivalContainer): Initializes artifact using a given univariate container.

plot() → matplotlib.figure.Figure: Artifact visualization.

get_path_to_plot(column_name: Optional[str] = None) → str: Returns a path where plot will be dumped.

get_path_to_artifact(column_name: Optional[str] = None) → str: Returns a path where artifact will be dumped.

save_plot(): Saves artifact’s plot.

dump(path: str): Dumps the object into a given pickle file.

static load(path: str) → logml.survival_analysis.artifacts.base.BaseArtifact: Loads KM artifacts from a given pickle object.