Foldbender

<a target="_blank" href="https://colab.research.google.com/github/branchialspace/foldbender/blob/main/foldbender.ipynb"> <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/> </a>

Functions for converting Alphafold PDB molecules into Pytorch Geometric graph representations for use with graph network models.

# Requirements (for Colab runtime)
# !pip install pyg-lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cu121.html > /dev/null 2>&1
!pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cpu.html > /dev/null 2>&1
!pip install torch-geometric > /dev/null 2>&1
!pip install rdkit > /dev/null 2>&1
!pip install Bio > /dev/null 2>&1
!echo 'debconf debconf/frontend select Noninteractive' | sudo debconf-set-selections > /dev/null 2>&1 && sudo apt-get install -y dssp > /dev/null 2>&1
!pip install ase > /dev/null 2>&1
!pip install dscribe > /dev/null 2>&1
!pip install iterative-stratification > /dev/null 2>&1
!git clone https://github.com/branchialspace/foldbender.git > /dev/null 2>&1

import foldbender as fb

# 1
fb.fasta_alpha(input_fasta=".fasta", input_dir="")

# 2
fb.alpha_nx(input_dir="", output_dir="")

# 3
fb.nx_pyg(input_dir="", output_dir="")

# 4
fb.soap_local(input_dir="", r_cut=3, n_max=3, l_max=3, sigma=0.1)

# 5 (gpu)
fb.precompute_eigens(input_dir, max_freqs=16)


# Task specific:

fb.minmax_norm(input_dir="")

fb.go_labels(input_dir="", train_terms=".tsv")

fb.go_split(input_dir="")

fb.esm2_labels(embeddings_path=".npy", sequence_ids_path=".npy", input_dir="")

fb.foldseek_targets(file_clusters=".tsv", file_scores=".tsv")

fb.foldseek_multiclass_labels(input_dir="", foldseek_targets=".tsv")

fb.foldseek_regression_labels(input_dir="", foldseek_targets=".tsv")

fb.foldseek_multiclass_split(input_dir="", valid_size=0.3, test_size=0.3, random_state=42)