文件名称:deep_cox_mixtures
文件大小:874KB
文件格式:ZIP
更新时间:2024-05-28 09:32:30
Python
深考克斯混合物 Chirag Nagpal 1,2 Steve Yadlowsky 1 ,Negar Rostamzadeh 1和Katherine Heller 1 1个Google脑部团队, 2个卡内基梅隆大学 该存储库包含NeurIPS 2020,ML4H论文的代码: 用法 from dcm import deep_cox_mixtures results = deep_cox_mixture.experiment(dataset='SUPPORT', prot_att='race', groups=('white', 'other')) deep_cox_mixtures.display_results(results) 要求 dcm取决于tensorflow2和scikit-survival , 运行基线模型进行比较需要lifelines , pycox和dsm
【文件预览】:
deep_cox_mixtures-master
----dcm()
--------deep_cox_mixture.py(11KB)
--------data_utils.py(5KB)
--------calibration.py(9KB)
--------dcm_tf.py(13KB)
--------baseline_models.py(15KB)
--------models.py(5KB)
--------metrics.py(6KB)
--------skmetrics.py(25KB)
--------plots.py(14KB)
----LICENSE(581B)
----README.md(720B)
----data()
--------support2.csv(3MB)
--------flchain.csv(317KB)