class DTIDataset(data.Dataset): def __init__(self, list_IDs, df, max_drug_nodes=290): self.list_IDs = list_IDs self.df = df self.max_drug_nodes = max_drug_nodes self.atom_featurizer = CanonicalAtomFeaturizer() self.bond_featurizer = CanonicalBondFeaturizer(self_loop=True) self.fc = partial(smiles_to_bigraph, add_self_loop=True)是什么意思
时间: 2023-05-10 11:55:37 浏览: 148
souhu_data.zip_dataset_saohu.pubn_sohu_text document
这段代码定义了一个名为DTIDataset的类,它继承自data.Dataset。在初始化函数中,它接受三个参数:list_IDs、df和max_drug_nodes。它还定义了一个atom_featurizer和一个bond_featurizer,它们分别使用CanonicalAtomFeaturizer和CanonicalBondFeaturizer类进行初始化。最后,它定义了一个fc函数,它使用smiles_to_bigraph函数将SMILES字符串转换为图形表示,并添加了自环。
阅读全文