ADMET-AI: A machine learning ADMET platform for evaluation of large-scale chemical libraries – Data and Models

<p>This repository contains data and models used in the following paper.</p> <p>&nbsp;</p> <p>Swanson, K., Walther, P., Leitz, J., Mukherjee, S., Wu, J. C., Shivnaraine, R. V., &amp; Zou, J. ADMET-AI: A machine learning ADMET platform for evaluation of large-scale chemical libraries. In review.</p> <p>&nbsp;</p> <p>The data and models are meant to be used with the <a href="https://github.com/swansonk14/admet_ai">ADMET-AI</a> code, which runs the ADMET-AI web server at <a href="https://admet.ai.greenstonebio.com/">admet.ai.greenstonebio.com</a>.</p> <p>&nbsp;</p> <p>The data.zip file has the following structure.</p> <p>data</p> <p>&nbsp; drugbank: Contains files with drugs from&nbsp;the <a href="https://go.drugbank.com/">DrugBank</a> that have received regulatory approval. drugbank_approved.csv contains the full set of approved drugs along with ADMET-AI predictions, while the other files contain subsets of these molecules used for testing the speed of ADMET prediction tools.</p> <p>&nbsp; tdc_admet_all: Contains the data (.csv files) and RDKit features (.npz files) for all 41 single-task ADMET datasets from the <a href="https://tdcommons.ai/">Therapeutics Data Commons</a> (TDC).</p> <p>&nbsp; tdc_admet_multitask: Contains the data (.csv files) and RDKit features (.npz files) for the two multi-task datasets (one regression and one classification) constructed by combining the tdc_admet_all datasets.</p> <p>&nbsp; tdc_admet_all.csv: A CSV file containing all 41 ADMET datasets from tdc_admet_all. This can be used to easily look up all ADMET properties for a given molecule in the TDC.</p> <p>&nbsp; tdc_admet_group: Contains the data (.csv files) and RDKit features (.npz files) for the 22 TDC ADMET Benchmark Group datasets with five splits per dataset.</p> <p>&nbsp; tdc_admet_group_raw: Contains the raw data (.csv files) used to construct the five splits per dataset in tdc_admet_group.</p> <p>&nbsp;</p> <p>The models.zip file has the following structure. Note that the ADMET-AI website and Python package use the multi-task Chemprop-RDKit models below.</p> <p>models</p> <p>&nbsp; tdc_admet_all: Contains Chemprop and Chemprop-RDKit models trained on all 41 single-task TDC ADMET datasets.</p> <p>&nbsp; tdc_admet_all_multitask: Contains Chemprop and Chemprop-RDKit models trained on the two multi-task TDC ADMET datasets (one regression and one classification).</p> <p>&nbsp; tdc_admet_group: Contains Chemprop and Chemprop-RDKit models trained on the 22 TDC ADMET Benchmark Group datasets.</p>