【文件属性】:
文件名称:retain-keras:RETAIN递归神经网络在Keras中的重新实现
文件大小:32KB
文件格式:ZIP
更新时间:2021-05-17 01:59:29
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)