AttentionAML: An Attention-based Deep Learning Framework for Accurate Molecular Diagnosis and Prognosis of Acute Myeloid Leukemia
AttentionAML (an Attention-Based Deep Learning Framework for Identifying Acute Myeloid Leukemia Subtypes), an accurate and effective model for AML subtype identification.
- Clone the AttentionAML git repository
git clone https://github.com/wan-mlab/AttentionAML.git- Navigate to the directory of AttentionAML package
cd /your path/AttentionAML
pip install .- Modify the System Path and import module
import sys; sys.path.append('AttentionAML')
from AttentionAML import AttentionAML- unzip and read the test file
test = pd.read_csv('TPM_test.csv', index_col=0)- AML subtype prediction
AttentionAML.Predict(Exp = test, exp_type = 'TPM')- AML risk prediction
risk_efs = AttentionAML.PredictRisk(Exp = test, exp_type = 'TPM', endpoint = 'efs')
risk_os = AttentionAML.PredictRisk(Exp = test, exp_type = 'TPM', endpoint = 'os')- Example Outputs
The prediction results will be stored and exported to the Prediction_results.csv
Lusheng Li, Shibiao Wan
AttentionAML: An Attention-based Deep Learning Framework for Accurate Molecular Categorization of Acute Myeloid Leukemia Lusheng Li, Joseph D. Khoury, Jieqiong Wang, Shibiao Wan bioRxiv 2025.05.20.655179v1; doi: https://doi.org/10.1101/2025.05.20.655179
GNU GENERAL PUBLIC LICENSE
Version 3, 29 June 2007
This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with this program. If not, see https://www.gnu.org/licenses/.