graph LR
Model_Core["Model Core"]
Training_and_Prediction_Orchestrator["Training and Prediction Orchestrator"]
Data_Management["Data Management"]
Evaluation_and_Visualization["Evaluation and Visualization"]
Model_Interpretation["Model Interpretation"]
Training_and_Prediction_Orchestrator -- "initializes" --> Model_Core
Training_and_Prediction_Orchestrator -- "uses" --> Model_Core
Training_and_Prediction_Orchestrator -- "loads data from" --> Data_Management
Model_Interpretation -- "extracts data from" --> Data_Management
Model_Interpretation -- "interacts with" --> Model_Core
Evaluation_and_Visualization -- "receives predictions from" --> Training_and_Prediction_Orchestrator
click Model_Core href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/decima/Model Core.md" "Details"
click Training_and_Prediction_Orchestrator href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/decima/Training and Prediction Orchestrator.md" "Details"
click Data_Management href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/decima/Data Management.md" "Details"
click Evaluation_and_Visualization href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/decima/Evaluation and Visualization.md" "Details"
click Model_Interpretation href "https://github.com/CodeBoarding/GeneratedOnBoardings/blob/main/decima/Model Interpretation.md" "Details"
The decima project implements a neural network model for analyzing biological sequence and gene expression data. Its core functionality involves training a deep learning model, managing complex HDF5 datasets, and providing tools for model evaluation, visualization, and interpretation of predictions.
Encapsulates the fundamental neural network architecture (DecimaModel), the custom loss function (TaskWisePoissonMultinomialLoss), and the disease-specific evaluation metric (DiseaseLfcMSE). It forms the computational backbone of the Decima model.
Related Classes/Methods:
decima.src.decima.decima_model.DecimaModel(11:53)decima.src.decima.loss.TaskWisePoissonMultinomialLoss(6:41)decima.src.decima.metrics.DiseaseLfcMSE(6:27)
Integrates the Decima model with the PyTorch Lightning framework, managing the entire machine learning lifecycle including model initialization, forward passes, training, validation, testing, and prediction steps. It handles data loading and orchestrates the training and prediction processes.
Related Classes/Methods:
Responsible for reading, processing, and augmenting biological sequence and gene expression data stored in HDF5 files. It includes utilities for gene indexing, extracting specific data points, and defining dataset classes for efficient data loading during training and inference, including specific handling for variant data.
Related Classes/Methods:
decima.src.decima.read_hdf5.HDF5Dataset(74:157)decima.src.decima.read_hdf5.VariantDataset(160:249)decima.src.decima.read_hdf5.get_gene_idx(39:41)decima.src.decima.read_hdf5.list_genes(30:36)decima.src.decima.read_hdf5.extract_gene_data(50:64)decima.src.decima.read_hdf5._extract_center(44:47)decima.src.decima.read_hdf5.mutate(67:71)decima.src.decima.preprocess.make_inputs(268:281)
Provides functionalities for quantitatively assessing the performance of the model, particularly focusing on marker gene analysis and criteria matching, and generates plots and visual representations of the model's evaluation results.
Related Classes/Methods:
decima.src.decima.evaluate.compare_marker_zscores(33:37)decima.src.decima.evaluate.marker_zscores(20:30)decima.src.decima.evaluate.match_criteria(8:17)decima.src.decima.evaluate.compute_marker_metrics(40:71)decima.src.decima.visualize.plot_marker_box(116:201)decima.src.decima.visualize.plot_logo(9:15)decima.src.decima.visualize.plot_gene_scatter(18:62)decima.src.decima.visualize.plot_track_scatter(65:113)decima.src.decima.visualize.plot_attribution_peaks(204:218)
Provides tools and methods for interpreting the predictions and internal workings of the model, such as attribution analysis to understand feature importance and motif scanning.
Related Classes/Methods: