本库用于存储关于 neural ODE 的相关代码和文档。主要分为三个部分:
- 使用 neural ODE 进行简单时间序列的动力学学习
- 使用 neural ODE 实现连续化的标准化流 (normalizing flow)
- 将 Zhang, 2022 中模型的各部分替换为 neural ODE,使用 neural ODE 进行心率信号的分类
详细见ode.ipynb文件
normalizing flow
文件夹中存放了使用 neural ODE 实现的标准化流的代码utils
文件夹中存放了一些用于训练工具函数,如数据集的生成,模型的保存等
PlanarFlow_training.ipynb
中构建了一个 planar flow 的模型,并进行训练PlanarFlow_validation.ipynb
中使用训练好的 planar flow 模型进行验证,所有的模型参数都保存在Planar
文件夹中
Phi.py
中定义了 OT flow 中所需要的 Phi 函数,以及 Phi 函数的梯度以及 Hessian 矩阵的迹。OTFlow_training.ipynb
中构建了一个 OT flow 的模型,并进行训练OTFlow_validation.ipynb
中使用训练好的 OT flow 模型进行验证,所有的模型参数都保存在OT-Flow
文件夹中