PI-CVAE 文件夹说明

本文件夹包含所有 PI-CVAE (Physics-Informed Conditional Variational Autoencoder) 相关的代码文件。

文件结构

核心模型文件

• pi_cvae_model.py - 模型定义（双分支编码器、解码器）
• pi_cvae_losses.py - 物理损失函数

训练和测试脚本

• pi_cvae_train.py - 通用训练脚本
• pi_cvae_train_example.py - 使用原始数据格式的训练示例
• pi_cvae_test.py - 测试和不确定性传播脚本

工具文件

• data_utils.py - 数据加载和预处理工具
• visualization.py - 可视化工具

示例和快速启动

• example_usage.py - 完整使用示例
• quick_start.py - 快速测试脚本

文档

• README_PI_CVAE.md - 详细使用说明
• PROJECT_SUMMARY.md - 项目文件说明
• requirements.txt - Python依赖包列表

使用方法

从父目录运行

# 训练模型
python pi_cvae/pi_cvae_train.py --use_synthetic

# 使用原始数据格式训练
python pi_cvae/pi_cvae_train_example.py

# 测试模型
python pi_cvae/pi_cvae_test.py --checkpoint checkpoints/best_model.pth

在文件夹内运行

cd pi_cvae

# 快速测试
python quick_start.py

# 完整示例
python example_usage.py

导入模块

如果要在其他脚本中使用PI-CVAE模块：

# 方式1：从pi_cvae包导入
from pi_cvae import PI_CVAE, PhysicsInformedLoss

# 方式2：直接导入模块
from pi_cvae.pi_cvae_model import PI_CVAE
from pi_cvae.data_utils import load_frf_data

注意事项

• 所有文件之间的相对导入保持不变，因为它们都在同一个文件夹中
• 如果从外部调用，确保Python路径包含父目录
• 数据文件路径需要根据实际情况调整