PHOTONAI-A Python API for rapid machine learning model development

PLoS One. 2021 Jul 21;16(7):e0254062. doi: 10.1371/journal.pone.0254062. eCollection 2021.

Abstract

PHOTONAI is a high-level Python API designed to simplify and accelerate machine learning model development. It functions as a unifying framework allowing the user to easily access and combine algorithms from different toolboxes into custom algorithm sequences. It is especially designed to support the iterative model development process and automates the repetitive training, hyperparameter optimization and evaluation tasks. Importantly, the workflow ensures unbiased performance estimates while still allowing the user to fully customize the machine learning analysis. PHOTONAI extends existing solutions with a novel pipeline implementation supporting more complex data streams, feature combinations, and algorithm selection. Metrics and results can be conveniently visualized using the PHOTONAI Explorer and predictive models are shareable in a standardized format for further external validation or application. A growing add-on ecosystem allows researchers to offer data modality specific algorithms to the community and enhance machine learning in the areas of the life sciences. Its practical utility is demonstrated on an exemplary medical machine learning problem, achieving a state-of-the-art solution in few lines of code. Source code is publicly available on Github, while examples and documentation can be found at www.photon-ai.com.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Datasets as Topic
  • Machine Learning*
  • Neural Networks, Computer
  • Software*
  • Workflow

Grants and funding

This work was supported by grants from the Interdisciplinary Center for Clinical Research (IZKF, https://www.medizin.uni-muenster.de/izkf.html) of the medical faculty of Münster (grant MzH 3/020/20 to TH and grant Dan3/012/17 to UD) and the German Research Foundation (DFG, https://www.dfg.de/, grants HA7070/2-2, HA7070/3, HA7070/4 to TH). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.