torchcell.viz
Plotting helpers used during training and analysis: predicted-versus-measured fitness and genetic-interaction plots, dataset split visualizations, graph-regularization and edge-recovery plots, transformer diagnostics, oversmoothing and oversquashing measures, and regression diagnostics logged to Weights & Biases.
fitness
Box-plot visualization of predicted vs measured growth/fitness.
Functions
Bin predictions into growth ranges and box-plot the measured values. |
|
Generate correlated synthetic true/predicted growth values with WT samples. |
|
Generate synthetic growth data seeded with NaN, inf, and identical-value bins. |
genetic_interaction_score
Visualizations of genetic interaction score predictions versus truth.
Functions
Build a binned box-plot figure comparing predictions to true scores. |
|
Generate synthetic true values and noisy predictions for plot testing. |
|
Generate synthetic data with injected NaN values for robustness testing. |
graph_recovery
Plotting utilities for graph regularization and edge-recovery metrics.
Classes
Visualization class for graph regularization and edge recovery metrics. |
transformer_diagnostics
Visualization helpers for transformer-specific diagnostic metrics.
Classes
Visualization class for transformer-specific diagnostic metrics. |
visual_graph_degen
Graph degeneration diagnostics: oversmoothing and oversquashing metrics.
Classes
Compute and log graph oversmoothing/oversquashing diagnostics to wandb. |
visual_regression
Visualization helpers for logging regression diagnostics to Weights & Biases.
Classes
Create and log regression diagnostic figures to W&B. |