Non-parametric lifetime models

A built-in dataset of circuit breaker lifetimes.

from relife.datasets import load_circuit_breaker

dataset = load_circuit_breaker()
print(dataset["time"])
print(dataset["event"])
print(dataset["entry"])

[34. 28. 12. ... 42. 42. 37.]
[ True  True  True ... False False False]
[33. 27. 11. ... 31. 31. 26.]

from relife.lifetime_models import KaplanMeier, NelsonAalen

kaplan_meier = KaplanMeier()
kaplan_meier.fit(
    dataset["time"],
    event=dataset["event"],
    entry=dataset["entry"],
)

<relife.lifetime_models._non_parametric_models.KaplanMeier at 0x7dff59d3c990>

nelson_aalen = NelsonAalen()
nelson_aalen.fit(
    dataset["time"],
    event=dataset["event"],
    entry=dataset["entry"],
)

<relife.lifetime_models._non_parametric_models.NelsonAalen at 0x7dff59d4b2d0>

import matplotlib.pyplot as plt

fig, axs = plt.subplots(ncols=2, nrows=1, figsize=(10, 4))
kaplan_meier.plot("sf", ax=axs[0])
nelson_aalen.plot("chf", ax=axs[1])
axs[0].set_title("Kaplan-Meier")
axs[1].set_title("Nelson-Aalen")
plt.show()