heat1d-pde-dataset
收藏heat1d-pde-dataset
数据集描述
数据集概述
该数据集包含带有冷却项的1D热方程的数值解,专为科学计算和物理信息神经网络的机器学习应用而设计。每个样本包括初始状态、最终状态(带噪声和不带噪声)、模拟参数和经过的时间。
支持的任务
- PDE解预测
- 参数推断
- 物理信息机器学习
- 科学机器学习基准测试
数据集结构
{ initial_states: [N, 200], # 初始温度分布 final_states: [N, 200], # 最终温度分布(带噪声) clean_initial_states: [N, 200], # 无噪声的初始状态 clean_final_states: [N, 200], # 无噪声的最终状态 parameters: [N, 3], # [alpha, k, t_env] elapsed_times: [N], # 初始状态和最终状态之间的时间 }
数据字段
initial_states: t=0时的温度分布final_states: t=elapsed_time时的温度分布clean_initial_states: 无噪声的初始状态clean_final_states: 无噪声的最终状态parameters:alpha: 热扩散率 [1e-5, 1e-4]k: 冷却系数 [0.01, 0.1]t_env: 环境温度 [15, 35]
elapsed_times: 状态之间的时间差
数据分割
所有数据都提供在训练集中。用户应创建自己的验证/测试分割。
源代码
数据集使用有限差分法求解热方程生成:
∂T/∂t = α∂²T/∂x² - k(T - T_env)
边界条件:
- T(x=0, t) = temp1
- T(x=L, t) = temp2
噪声水平
- 输入状态:温度范围的1%
- 输出状态:温度范围的0.5%
- 参数:参数值的1%
使用方法
安装datasets库: bash pip install datasets
加载数据集: python from datasets import load_dataset
本地下载文件
dataset = load_dataset("nick-leland/heat1d-pde-dataset", download_mode="force_redownload")
读取初始结构(h5py文件)
df = dataset[train].data.to_pandas() file_path = df[image][0][path] data = h5py.File(file_path, r)
访问数据
initial_states = data[initial_states][:] final_states = data[final_states][:] parameters = data[parameters][:] elapsed_times = data[elapsed_times][:]
数据集创建者
Nicholas Leland
许可信息
MIT许可证




