文件名称:retain-keras:RETAIN递归神经网络在Keras中的重新实现
文件大小:32KB
文件格式:ZIP
更新时间:2024-06-05 19:46:09
Python
RETAIN-Keras:重新实现RETAIN的Keras ,它允许创建高度可解释的递归神经网络模型以进行患者诊断,而不会降低模型性能 该存储库保留了Keras对RETAIN的重新实现,可以灵活地修改原始代码,引入了多个新功能,并提高了培训速度 RETAIN已证明对于创建多种情况的预测模型非常有效,我们很高兴与更广泛的医疗数据科学界分享这一实施方案 改进和附加功能 带有Tensorflow后端的更简单的Keras代码 能够使用固定大小的额外数字输入,该数字输入可以保存有关患者就诊的数字信息,例如患者的年龄,处方药的数量或血压 改进的嵌入逻辑,避免使用大型密集输入 训练期间评估模型的能力 仅需正面贡献即可训练模型的能力,从而提高了性能 额外的脚本可以评估模型并输出几个辅助图形 运行RETAIN-Keras 要求 的Python 3 如果使用GPU:CUDA和CuDNN Tensorfl
【文件预览】:
retain-keras-master
----.gitignore(10B)
----Notice.txt(928B)
----retain_train.py(20KB)
----LICENSE(11KB)
----process_mimic_modified.py(9KB)
----README.md(8KB)
----retain_evaluation.py(14KB)
----docs()
--------CONTRIBUTING.md(4KB)
--------CODE_OF_CONDUCT.md(3KB)
--------Keras-LICENSE.txt(2KB)
--------RETAIN-LICENSE.txt(1KB)
--------INDIVIDUAL_CONTRIBUTOR_LICENSE.md(6KB)
----retain_interpretations.py(13KB)