用数据集创建一个图，以便在其上使用 GNN

Question

我有包含每种材料的材料 ID、化学式和热电特性的数据集，我想使用 GNN 来预测一种热电特性。

首先，我为数据集的所有分子创建了一个结构。 现在想为我的数据集创建一个图表，包括节点属性和键属性。 但我不知道。

这是获取结构的代码

从 pymatgen.ext.matproj 导入 MPRester

import pandas as pd

api_key = "this was my api"

structures = []

with MPRester(api_key) as m:

 for material_id in df['mpid']:

structure = m.get_structure_by_material_id(material_id)

structures.append(structure)

data = {"material_id": materials, "structure": structures}

df1 = pd.DataFrame(data)