生存分析

一个实验性的生存分析库，最初受 Python 的 lifelines 启发。

当前限制

• 仅支持拟合 Weibull 分布
• 仅支持可以不通过导数进行拟合的分布
• 不支持自动微分

拟合数据

use ndarray::prelude::*;
use survival_analysis::{IntervalCensored, Fitter, BaseFitter};
use survival_analysis::distribution::WeibullDistribution;

let data = IntervalCensored {
    start: array![0., 2., 5., 10.],
    stop: array![2., 5., 10., 1e10f64],
};

let fitter = BaseFitter::new(data);

let params: WeibullDistribution<f64> = fitter.fit().unwrap();

assert!((params.shape - 0.980).abs() < 1e-2);
assert!((params.scale - 7.187).abs() < 1e-2);