Lightlines
By k-dense-ai

torchdrug

k-dense-ai1.7k installs

Build graph-learning workflows with TorchDrug.

What it does

Build and troubleshoot molecular, protein, and knowledge-graph learning workflows with TorchDrug.

When to use it

Use when code imports torchdrug or needs its datasets, models, tasks, or Engine.

How to use it

It changes how the agent builds and debugs TorchDrug code by checking compatibility and following documented dataset, model, task, and Engine workflows.

What you provide

  • An existing project

Uses

Good to know

  • advertises something the task does not need

Latest review

What you need · 8

Requires Python 3.7 through 3.10; newer Python combinations must not be presented as supported.

Requires PyTorch 1.8 through 2.0. Apple Silicon requires PyTorch 1.13 or later and CPU execution; MPS is unsupported.


Access · 5

This skill

TorchDrug documentation

Read

Consults TorchDrug dataset documentation

https://deepchemdata.s3-us-west-1.amazonaws.com/datasets/clintox.csv.gz

Read

Downloads the ClinTox dataset