Skip to content

Protein Dataset¤

This page is retained for historical links.

There is no artifex.data.protein_dataset module in the live runtime. The canonical owner is artifex.data.protein.dataset, re-exported through artifex.data.protein.

Current Imports¤

from artifex.data.protein import ProteinDataset, ProteinDatasetConfig

config = ProteinDatasetConfig(max_seq_length=128)
dataset = ProteinDataset(config, data_dir="./protein-pickles")

Use these current helpers from the canonical module:

  • ProteinDataset
  • ProteinDatasetConfig
  • ProteinStructure
  • protein_collate_fn
  • create_synthetic_protein_dataset
  • pdb_to_protein_example

ProteinDataset supports Datarax indexed batching through get_batch_at(start, size, key) and Pipeline(...). Those batches are fixed-shape tensor dictionaries padded to ProteinDatasetConfig.max_seq_length; use get_batch(...) or protein_collate_fn for local protein collation.

See Protein Dataset Module for the maintained reference page.