AbFlow

wenda8759/AbFlow

AbFlow is a graph machine learning model from wenda8759. Use it for the graph machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.

All datasets and pre-trained model weights are available on Hugging Face:

Downloads · 30 days

Access

Public

Updated Dec 29, 2025

Repo size

1.9 GB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.zip951 MB · 100%

At a glance

Task: Graph Machine Learning
License: cc-by-4.0
Access: Public
Created: Dec 22, 2025
Updated: Dec 29, 2025
SHA: 5f10cb52

Model card

From the Hugging Face model README

Data and Model Weights

All datasets and pre-trained model weights are available on Hugging Face:

🤗 https://huggingface.co/wenda8759/AbFlow

Download Options

Option 1: Using huggingface-cli (Recommended)

# Install huggingface_hub if not already installed
pip install huggingface_hub

# Download all files
huggingface-cli download wenda8759/AbFlow --local-dir ./

# Or download specific files
huggingface-cli download wenda8759/AbFlow checkpoints/multi_cdr_design.ckpt --local-dir ./

Option 2: Using Python

from huggingface_hub import snapshot_download, hf_hub_download

# Download entire repository
snapshot_download(repo_id="wenda8759/AbFlow", local_dir="./")

# Or download specific file
hf_hub_download(
    repo_id="wenda8759/AbFlow",
    filename="checkpoints/multi_cdr_design.ckpt",
    local_dir="./"
)

Model Checkpoints

Model	Description	File
Paratope-Centric Design	Design based on epitope	`checkpoints/paratope_centric_design.ckpt`
Multi-CDR Design	Design all 6 CDR regions	`checkpoints/multi_cdr_design.ckpt`
Structure Prediction	Predict antibody structure	`checkpoints/structure_prediction.ckpt`
Affinity Optimization	Optimize binding affinity	`checkpoints/affinity_optimization.ckpt`
ΔΔG Predictor	Predict binding energy changes	`checkpoints/ddg_predictor.ckpt`

Dataset Structure

After downloading, organize your data as follows:

AbFlow/
├── datasets/
│   ├── RAbD/
│   │   ├── train.json
│   │   ├── valid.json
│   │   ├── test.json
│   │   ├── train.pkl
│   │   ├── valid.pkl
│   │   ├── test.pkl
│   │   ├── train_surf.pkl
│   │   ├── valid_surf.pkl
│   │   └── test_surf.pkl
│   └── IgFold/
│       ├── train.json
│       ├── valid.json
│       ├── test.json
│       └── ...
└── checkpoints/
    ├── multi_cdr_design.ckpt
    ├── structure_prediction.ckpt
    └── ...

Task: Graph Machine Learning
License: cc-by-4.0
Created: Dec 22, 2025
Updated: Dec 29, 2025