Protein Dataset

Module: benchmarks.datasets.protein_dataset

Source: benchmarks/datasets/protein_dataset.py

Overview

Protein datasets for benchmarking.

This module provides synthetic protein datasets for benchmarking protein generative models.

Classes

SyntheticProteinDataset

class SyntheticProteinDataset

Functions

init

def __init__()

create_synthetic_protein_dataset

def create_synthetic_protein_dataset()

flatten_coordinates

def flatten_coordinates()

get_batch

def get_batch()

get_dataset_info

def get_dataset_info()

get_raw_coordinates

def get_raw_coordinates()

get_sample_with_metadata

def get_sample_with_metadata()

Module Statistics

• Classes: 1
• Functions: 7
• Imports: 6