可以使用模型的代码库中已经整理好的数据集:
https://widgets.figshare.com/articles/12522524/embed?show_title=1
也可使用Material Project数据集:
a.https://github.com/materialsvirtuallab/megnet/tree/master/mvl_models/mp-2019.4.1
b.https://github.com/materialsvirtuallab/megnet/tree/master/mvl_models/mp-2018.6.1
c.在线数据库
数据集不限于上述所述,开发者也可尝试使用其他数据库。数据集需自行拆分为训练集、验证集和测试集。训练集和验证集将包含分子/材料的结构信息及其已知属性,用于模型的训练和调参;测试集仅提供结构信息,用于最终评估模型的预测性能。