Non-parametric lifetime models

A built-in dataset of circuit breaker lifetimes.

from relife.data import load_circuit_breaker

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

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

from relife.lifetime_model import KaplanMeier, NelsonAalen

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

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

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])
plt.show()