用途:DeepChem 是用于化学、材料科学与生物学的 Python 机器学习库,支持分子性质预测、药物发现、材料设计与生物分子分析。
主要能力:提供多种数据加载器(CSVLoader、SDFLoader、FASTALoader、ImageLoader、JsonLoader);多种特征化方法(CircularFingerprint、RDKitDescriptors、MordredDescriptors、MolGraphConvFeaturizer、DMPNNFeaturizer、GroverFeaturizer、SmilesToSeq、CoulombMatrix 等);数据集划分器(ScaffoldSplitter、ButinaSplitter、MaxMinSplitter、RandomSplitter、RandomStratifiedSplitter);模型涵盖 SklearnModel、GBDTModel、MultitaskRegressor、GCNModel、GATModel、AttentiveFPModel、DMPNNModel、CGCNNModel、MEGNetModel、HuggingFaceModel、GroverModel 等;内置 MoleculeNet 基准数据集(Tox21、BBBP、HIV、ClinTox、Delaney、FreeSolv、Lipo、QM7/8/9、Perovskite、Bandgap 等);支持迁移学习与模型评估指标。
使用方式:安装命令为 uv pip install deepchem,PyTorch 模型用 uv pip install deepchem[torch],全部功能用 uv pip install deepchem[all]。典型流程为加载数据、特征化、划分数据集、训练模型、评估与预测。技能附带三个脚本:predict_solubility.py、graph_neural_network.py、transfer_learning.py。
适用场景:分子性质与 ADMET/毒性预测、MoleculeNet 基准评测、图神经网络建模、小数据集迁移学习、晶体与材料性质预测、蛋白或 DNA 序列分析。
限制与注意:分子任务建议使用 ScaffoldSplitter 防止数据泄漏;GNN 通常需要较大数据集(约 1 万样本以上);小数据集易过拟合,可用正则化、简单模型或迁移学习;大数据集可用 DiskDataset 并减小 batch size。具体收费、价格与平台信息暂未提供。