API

Preprocessing

Datasets

datasets.SelexDataset(df[, n_rounds, ...])

datasets.PBMDataset(df[, ...])

datasets.ResiduePBMDataset(df, msa_onehot[, ...])

datasets.GenomicsDataset(df[, ...])

Tools

Plotting

pl.logo(model[, figsize, flip, log, mode, ...])

pl.loss(model)

pl.kmer_enrichment(model, train[, k, ...])

pl.R2_per_protein(model, dataloader, device)

pl.R2_calculation(model, train[, show])

Models

models.Mubind(datatype, **kwargs)

Implements the MUBIND model.