PHOTONAI-A Python API for rapid machine learning model development

Ramona Leenings, Nils Ralf Winter, Lucas Plagwitz, Vincent Holstein, Jan Ernsting, Kelvin Sarink, Lukas Fisch, Jakob Steenweg, Leon Kleine- Vennekate, Julian Gebker, Daniel Emden, Dominik Grotegerd, Nils Opel, Benjamin Risse, Xiaoyi Jiang, Udo Dannlowski, Tim Hahn

Research output: Contribution to journalArticlepeer-review

22 Citations (Scopus)

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.

Original languageEnglish
Article numbere0254062
JournalPLoS One
Volume16
Issue numberJuly
DOIs
Publication statusPublished - Jul 2021
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2021 Leenings et al. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.

ASJC Scopus Subject Areas

  • General

PubMed: MeSH publication types

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

Fingerprint

Dive into the research topics of 'PHOTONAI-A Python API for rapid machine learning model development'. Together they form a unique fingerprint.

Cite this

Leenings, R., Winter, N. R., Plagwitz, L., Holstein, V., Ernsting, J., Sarink, K., Fisch, L., Steenweg, J., Kleine- Vennekate, L., Gebker, J., Emden, D., Grotegerd, D., Opel, N., Risse, B., Jiang, X., Dannlowski, U., & Hahn, T. (2021). PHOTONAI-A Python API for rapid machine learning model development. PLoS One, 16(July), Article e0254062. https://doi.org/10.1371/journal.pone.0254062