技能 torchdrug
⚗️

torchdrug

安全 ⚡ 包含脚本⚙️ 外部命令🌐 网络访问

使用 TorchDrug 和图神经网络进行药物发现

也可从以下获取: davila7

构建和训练用于药物发现、蛋白质建模和分子性质预测的图神经网络。本技能提供 TorchDrug 库的完整文档,包括数据集、模型架构和工作流程。

支持: Claude Codex Code(CC)
📊 71 充足
1

下载技能 ZIP

2

在 Claude 中上传

前往 设置 → 功能 → 技能 → 上传技能

3

开启并开始使用

测试它

正在使用“torchdrug”。 How do I train a model to predict HIV inhibition?

预期结果:

  • Load the HIV dataset from TorchDrug: datasets.HIV()
  • Use a GIN model for molecular graph representation
  • Create a PropertyPrediction task with binary classification
  • Train using BCE loss and evaluate with AUROC

正在使用“torchdrug”。 What datasets are available for protein function prediction?

预期结果:

  • EnzymeCommission for EC number classification across 7 levels
  • GeneOntology for GO term prediction (BP/MF/CC)
  • BetaLactamase for enzyme activity regression
  • Fluorescence for GFP protein intensity prediction

正在使用“torchdrug”。 How can I generate new drug-like molecules?

预期结果:

  • Use GCPN model for reinforcement learning-based generation
  • Apply GraphAutoregressiveFlow for conditional generation
  • Set property constraints like logP and synthesizability
  • Validate outputs with RDKit for chemical validity

安全审计

安全
v4 • 1/17/2026

All 335 static findings are FALSE POSITIVES. The skill contains only markdown documentation for TorchDrug, a legitimate PyTorch-based ML library for drug discovery. Security patterns detected are misidentified scientific terminology: PyTorch model methods (eval) flagged as code evaluation, markdown code block syntax (backticks) flagged as shell execution, ML loss functions (bce, mse) flagged as cryptographic algorithms, dataset names (SAMPL, ZINC, BindingDB) flagged as C2/SAM infrastructure. No executable code or security risks present.

10
已扫描文件
4,856
分析行数
3
发现项
4
审计总数
审计者: claude 查看审计历史 →

质量评分

45
架构
100
可维护性
85
内容
20
社区
100
安全
91
规范符合性

你能构建什么

预测分子性质

使用 GIN 和 GAT 等 GNN 架构预测溶解度、毒性和结合亲和力

建模蛋白质结构

使用 ESM 和 GearNet 模型分析蛋白质序列和结构以进行功能预测

规划合成路线

使用逆合成规划为目标分子设计化学合成途径

试试这些提示

快速入门
How do I install TorchDrug and run a basic example for molecular property prediction?
数据集选择
Which TorchDrug dataset should I use for training a model to predict blood-brain barrier penetration?
模型架构
What are the differences between GIN, GAT, and SchNet models in TorchDrug and when should I use each?
集成指南
How do I integrate TorchDrug with PyTorch Lightning for distributed training of large-scale molecular models?

最佳实践

  • 对分子数据集使用骨架分割以避免数据泄漏
  • 在扩展到更大的数据集之前,先使用 BACE 或 ESOL 等小型数据集
  • 将性质预测与生成模型结合进行多目标优化

避免

  • 不要对分子性质预测使用随机分割 - 骨架分割更真实
  • 避免在没有适当验证指标的情况下训练,对于不平衡数据集应使用 AUROC 和 AUPRC
  • 生成新分子时不要跳过 RDKit 验证

常见问题

TorchDrug 与 DeepChem 有什么不同?
TorchDrug 专注于 PyTorch 原生实现和自定义模型开发,而 DeepChem 则强调预训练模型和多样化的特征化方法。
哪些数据集最适合初学者?
从 BBBP、BACE 或 ESOL 数据集开始。它们体积小、文档完善,涵盖常见的药物发现任务。
TorchDrug 能处理蛋白质结构数据吗?
是的,TorchDrug 支持 AlphaFold PDB 文件和 ESM 嵌入,用于蛋白质序列和结构建模。
如何生成新的类药物分子?
使用 GCPN 或 GraphAutoregressiveFlow 模型并设置性质约束来引导分子生成。
有哪些可用的 GNN 架构?
GIN、GAT、GCN、RGCN、SchNet、GearNet、TransE、RotatE 和 ComplEx,适用于不同的数据类型和任务。
这个技能可以运行实际的模型训练吗?
不,本技能提供文档和参考指导。实际训练需要安装 TorchDrug 并运行 Python 代码。