PHOTONAI is a high level python API for designing and optimizing machine learning pipelines.
We've created a system in which you can easily select and combine both pre-processing and learning algorithms from state-of-the-art machine learning toolboxes, and arrange them in simple or parallel pipeline data streams.
In addition, you can parametrize your training and testing workflow choosing cross-validation schemes, performance metrics and hyperparameter optimization metrics from a list of pre-registered options.
Importantly, you can integrate custom solutions into your data processing pipeline, but also for any part of the model training and evaluation process including custom hyperparameter optimization strategies.
For a detailed description, visit our website and read the documentation
or you can read our paper in PLOS ONE
Getting Started
In order to use PHOTONAI you only need to have your favourite Python IDE ready. Then install the latest stable version simply via pip
pip install photonai
# Or try out the latest features if you don't rely on a stable version, using:
pip install --upgrade git+https://github.com/wwu-mmll/[email protected]
You can setup a full stack machine learning pipeline in a few lines of code:
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import KFold
from photonai.base import Hyperpipe, PipelineElement
from photonai.optimization import FloatRange, Categorical, IntegerRange
# DESIGN YOUR PIPELINE
my_pipe = Hyperpipe('basic_svm_pipe', # the name of your pipeline
# which optimizer PHOTONAI shall use
optimizer='sk_opt',
optimizer_params={'n_configurations': 25},
# the performance metrics of your interest
metrics=['accuracy', 'precision', 'recall', 'balanced_accuracy'],
# after hyperparameter optimization, this metric declares the winner config
best_config_metric='accuracy',
# repeat hyperparameter optimization three times
outer_cv=KFold(n_splits=3),
# test each configuration five times respectively,
inner_cv=KFold(n_splits=5),
verbosity=1,
project_folder='./tmp/')
# first normalize all features
my_pipe.add(PipelineElement('StandardScaler'))
# then do feature selection using a PCA
my_pipe += PipelineElement('PCA',
hyperparameters={'n_components': IntegerRange(5, 20)},
test_disabled=True)
# engage and optimize the good old SVM for Classification
my_pipe += PipelineElement('SVC',
hyperparameters={'kernel': Categorical(['rbf', 'linear']),
'C': FloatRange(0.5, 2)}, gamma='scale')
# train pipeline
X, y = load_breast_cancer(return_X_y=True)
my_pipe.fit(X, y)
Features
Easy access to established ML implementations
We pre-registered diverse preprocessing and learning algorithms from state-of-the-art toolboxes e.g. scikit-learn, keras and imbalanced learn to rapidly build custom pipelines
Hyperparameter Optimization
With PHOTONAI you can seamlessly switch between diverse hyperparameter optimization strategies, such as (random) grid-search or bayesian optimization (scikit-optimize, smac3).
Extended ML Pipeline
You can build custom sequences of processing and learning algorithms with a simple syntax. PHOTONAI offers extended pipeline functionality such as parallel sequences, custom callbacks in-between pipeline elements, AND- and OR- Operations, as well as the possibility to flexibly position data augmentation, class balancing or learning algorithms anywhere in the pipeline.
Model Sharing
PHOTONAI provides a standardized format for sharing and loading optimized pipelines across platforms with only one line of code.
Automation
While you concentrate on selecting appropriate processing steps, learning algorithms, hyperparameters and training parameters, PHOTONAI automates the nested cross-validated optimization and evaluation loop for any custom pipeline.
Results Visualization
PHOTONAI comes with extensive logging of all information in the training, testing and hyperparameter optimization process. In addition, optimum performances and the hyperparameter optimization progress are visualized in the PHOTONAI Explorer.
6 Dec 29, 2022
2 Jan 04, 2022
28 Oct 20, 2022
60 Dec 24, 2022
856 Jan 06, 2023
50 Dec 29, 2022
23 Dec 09, 2022
7 Feb 15, 2022
1 Feb 05, 2022
5 Feb 17, 2022
2.7k Jan 07, 2023
1 Jan 19, 2022
18.6k Jan 01, 2023
142 Dec 30, 2022
8 Sep 03, 2022
24 Dec 17, 2022
23 Dec 07, 2022
202 Jan 04, 2023
498 Jan 03, 2023
97 Dec 30, 2022