Accessible, uniform protein property prediction with a scikit-learn based toolset AIDE

TL;DR

AIDE is a Python tool that simplifies and standardizes protein property prediction using machine learning, with support for both labeled and unlabeled data.

Contribution

AIDE introduces a modular, scikit-learn compatible API for accessible and reproducible protein property prediction.

Findings

AIDE provides a standardized API for integrating various zero-shot and supervised prediction methods.

The tool supports reproducible workflows for analyzing protein variants and homologs.

AIDE is compatible with scikit-learn transformers and pipelines for streamlined use.

Abstract

Protein property prediction via machine learning with and without labeled data is becoming increasingly powerful, yet methods are disparate and capabilities vary widely over applications. The software presented here, “Artificial Intelligence Driven protein Estimation (AIDE)”, enables instantiating, optimizing, and testing many zero-shot and supervised property prediction methods for variants and variable length homologs in a single, reproducible notebook or script by defining a modular, standardized application programming interface (API), i.e. drop-in compatible with scikit-learn transformers and pipelines.

AIDE is an installable, importable python package inheriting from scikit-learn classes and API and is installable on Windows, Mac, and Linux. Many of the wrapped models internal to AIDE will be effectively inaccessible without a GPU, and some assume CUDA. The newest stable, tested version can be found at https://github.com/beckham-lab/aide_predict and a full user guide and API reference can be found at https://beckham-lab.github.io/aide_predict/. Static versions of both at the time of writing can be found on Zenodo.

Figures