大原子模型时代的图神经网络数据集及处理脚本

数据集概述

本数据集为“大原子模型时代的图神经网络”研究提供支持，包含力场预测零样本测试数据集、单任务基准数据集、多任务预训练大原子模型文件及零样本测试复现脚本，覆盖数据、模型、脚本三类核心内容。

文件详解

• 压缩文件: dpa3_test_data_script.zip，格式为ZIP，包含以下目录结构及文件：
• data目录：含Force-Field-Task-Datasets（12个力场预测零样本测试数据集）和Single-Task-Datasets（需预处理的单任务基准数据集），详情见对应子目录README
• model目录：DPA-3.1-3M.pt（多任务预训练大原子模型文件）
• script目录：zero-shot-test/zero-shot-test.py（复现模型零样本测试结果的脚本）

适用场景

• 原子模型研究：验证图神经网络在大原子模型任务中的性能
• 零样本学习分析：基于力场预测数据集开展零样本测试方法研究
• 模型复现：使用预训练模型及脚本复现原研究实验结果
• 原子模拟算法开发：为单任务基准数据集的预处理与模型训练提供基础数据