Categorizing Pediatric Acute Myeloid Leukemia by a Multi-Omics Integration Ensemble Learning Model

AttentionAML (an Attention-Based MLP Model for Identifying Acute Myeloid Leukemia Subtypes), an accurate and effective model for AML subtype identification.

Flowchart of AttentionAML

Installation

1. Clone the AttentionAML git repository

git clone https://github.com/wan-mlab/AttentionAML.git

2. Navigate to the directory of AttentionAML package

cd /your path/AttentionAML
pip install .

Tutorial

Jupyter notebook

1. Modify the System Path and import module

import sys; sys.path.append('AttentionAML')
from AttentionAML import AttentionAML

2. unzip and read the test file

test = pd.read_csv('TPM_test.csv', index_col=0)

3. AML subtype prediction

AttentionAML.Predict(Exp = test, exp_type = 'TPM')

Authors

Lusheng Li, Shibiao Wan

Publication

AttentionAML: An attention-based deep learning model for accurate identification of childhood acute myeloid leukemia subtypes Lusheng Li, Joseph D. Khoury, Jieqiong Wang, Shibiao Wan Cancer Res (2025) (AACR Abstract)

Manuscript in progress