DeepChem的内置数据集及使用方法

最新推荐文章于 2025-09-11 07:07:14 发布

原创最新推荐文章于 2025-09-11 07:07:14 发布 · 1.5k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#神经网络 #深度学习 #机器学习 #人工智能

药物设计同时被 2 个专栏收录

60 篇文章

订阅专栏

图神经网络

38 篇文章

订阅专栏

本文介绍了DeepChem中内置的MoleculeNet数据集及其使用方法。MoleculeNet包含大量分子数据，目前共有46个数据集。通过示例展示了如何加载特定数据集并访问其特征、标签等。


'''

by wufeil

DeepChem的内置数据集及使用方法
Tutorial 3: An Introduction To MoleculeNet

Deepchem含有大量的内置数据集，主要是一些分子的数据，所以，数据集的名字为：MoleculeNet (分子网络)
数据集一直在更新，目前已经有46个数据集，

deepchem的数据集包含在molnet模块内, 使用load_+模块名称加载数据
如下：
'''

import deepchem as dc

tasks, datasets, transformer = dc.molnet.load_delaney(featurizer='GraphConv', splitter='random')
trainset, validset, testset = datasets
for X, y, w, ids in testset.itersamples():
    '''
    X-特征
    y-标签
    w-样本权重
    ids- ID
    
    注：
        datasets已经将训练集、验证集、测试集分开了
        数据集testset不是可以直接迭代的对象，要testset.itersamples()才可以
        具体请见：https://blog.csdn.net/wufeil7/article/details/110631024?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522160732181019721940220137%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=160732181019721940220137&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_v2~rank_v29-3-110631024.pc_v2_rank_blog_default&utm_term=deepchem&spm=1018.2118.3001.4450
    '''
    print(X, y, w, ids)