Protein Dataset

Module: data.protein_dataset

Source: data/protein_dataset.py

Overview

Protein dataset implementation for diffusion models.

This module provides dataset classes for loading, processing, and batching protein structure data for diffusion models.

Classes

ProteinDataset

class ProteinDataset

ProteinStructure

class ProteinStructure

Functions

init

def __init__()

collate_batch

def collate_batch()

create_synthetic_protein_dataset

def create_synthetic_protein_dataset()

from_numpy

def from_numpy()

get_batch

def get_batch()

get_statistics

def get_statistics()

Module Statistics

• Classes: 2
• Functions: 6
• Imports: 4