logml.survival_analysis.artifacts.cox

Classes

CoxArtifact(*args)

Object for handling Cox regression: summary + HR plot.

class logml.survival_analysis.artifacts.cox.CoxArtifact(*args)

Bases: logml.survival_analysis.artifacts.base.BaseArtifact

Object for handling Cox regression: summary + HR plot.

LABEL = 'cox'
check_significance()

Checks whether all inner covariates are significant.

select_least_significant_features() List[Tuple[str, float]]

Returns a few least significant features according to Cox.

build(container: logml.data.datasets.survival_dataset.SurvivalDataset)

Initializes artifact using a given survival dataset.

Does the following:
  • fits Cox model

  • applies FDR to covariates p-values

show_summary()

Shows inner Cox model summary.

plot() matplotlib.figure.Figure

Artifact visualization: HR plot.

column_name: str
threshold: float