package | |
---|---|
tests | |
docs | |
license | |
stats | |
support |
This repository contains FEDOT - an open-source framework for automated modeling and machine learning (AutoML). It can build custom modeling pipelines for different real-world processes in an automated way using an evolutionary approach. FEDOT supports classification (binary and multiclass), regression, clustering, and time series prediction tasks.
The main feature of the framework is the complex management of interactions between various blocks of pipelines. First of all, this includes the stage of machine learning model design. FEDOT allows you to not just choose the best type of the model, but to create a complex (composite) model. It allows you to combine several models of different complexity, which helps you to achieve better modeling quality than when using any of these models separately. Within the framework, we describe composite models in the form of a graph defining the connections between data preprocessing blocks and model blocks.
The framework is not limited to specific AutoML tasks (such as pre-processing of input data, feature selection, or optimization of model hyperparameters), but allows you to solve a more general structural learning problem - for a given data set, a solution is built in the form of a graph (DAG), the nodes of which are represented by ML models, pre-processing procedures, and data transformation.
The project is maintained by the research team of the Natural Systems Simulation Lab, which is a part of the National Center for Cognitive Research of ITMO University.
The intro video about Fedot is available here:
The main features of the framework are as follows:
- The FEDOT architecture is highly flexible and therefore the framework can be used to automate the creation of mathematical models for various problems, types of data, and models;
- FEDOT already supports popular ML libraries (scikit-learn, keras, statsmodels, etc.), but you can also integrate custom tools into the framework if necessary;
- Pipeline optimization algorithms are not tied to specific data types or tasks, but you can use special templates for a specific task class or data type (time series forecasting, NLP, tabular data, etc.) to increase the efficiency;
- The framework is not limited only to machine learning, it is possible to embed models related to specific areas into pipelines (for example, models in ODE or PDE);
- Additional methods for hyperparameters tuning can be seamlessly integrated into FEDOT (in addition to those already supported);
- The resulting pipelines can be exported in a human-readable JSON format, which allows you to achieve reproducibility of the experiments.
Thus, compared to other frameworks, FEDOT:
- Is not limited to specific modeling tasks and claims versatility and expandability;
- Allows managing the complexity of models and thereby achieving better results.
- Allows building models using input data of various nature (texts, images, tables, etc.) and consisting of different types of models.
Common installation:
$ pip install fedot
In order to work with FEDOT source code:
$ git clone https://github.com/nccr-itmo/FEDOT.git $ cd FEDOT $ pip install -r requirements.txt $ pytest -s test
FEDOT provides a high-level API that allows you to use its capabilities in a simple way. At the moment, the API can be used for classification and regression tasks only. But the time series forecasting and clustering support will be implemented soon (you can still solve these tasks via advanced initialization, see below). Input data must be either in NumPy arrays or CSV files.
To use the API, follow these steps:
- Import Fedot class
from fedot.api.main import Fedot
- Initialize the Fedot object and define the type of modeling problem. It provides a fit/predict interface:
- fedot.fit runs the optimization and returns the resulting composite model;
- fedot.predict returns the prediction for the given input data;
- fedot.get_metrics estimates the quality of predictions using selected metrics
Numpy arrays, pandas data frames, and file paths can be used as sources of input data.
model = Fedot(problem='classification')
model.fit(features=train_data.features, target=train_data.target)
prediction = model.predict(features=test_data.features)
metrics = model.get_metrics()
For more advanced approaches, please use Examples & Tutorials section.
Jupyter notebooks with tutorials are located in the examples repository. There you can find the following guides:
- Intro to AutoML
- Intro to FEDOT functionality
- Intro to time series forecasting with FEDOT
- Advanced time series forecasting
- Gap-filling in time series and out-of-sample forecasting
Notebooks are issued with the corresponding release versions (the default version is 'latest').
Also, external examples are available:
Extended examples:
- Credit scoring problem, i.e. binary classification task
- Time series forecasting, i.e. random process regression
- Spam detection, i.e. natural language preprocessing
- Movie rating prediction with multi-modal data
Also, several video tutorials are available (in Russian).
We also published several posts and news devoted to the different aspects of the framework:
In English:
- How AutoML helps to create composite AI? - towardsdatascience.com
- AutoML for time series: definitely a good idea - towardsdatascience.com
- AutoML for time series: advanced approaches with FEDOT framework - towardsdatascience.com
- Winning a flood-forecasting hackathon with hydrology and AutoML - towardsdatascience.com
- FEDOT as a factory of human-competitive results - youtube.com
In Russian:
- General concepts of evolutionary design for composite pipelines - habr.com
- Automated time series forecasting with FEDOT - habr.com
- Details of FEDOT-based solution for Emergency DataHack - habr.com
In Chinese:
- Presentation of FEDOT in Open innovations 2.0 conference - youtube.com
The latest stable release of FEDOT is on the master branch.
The repository includes the following directories:
- Package core contains the main classes and scripts. It is the core of FEDOT framework
- Package examples includes several how-to-use-cases where you can start to discover how FEDOT works
- All unit and integration tests can be observed in the test directory
- The sources of the documentation are in the docs
Also, you can check benchmarking a repository that was developed to provide a comparison of FEDOT against some well-known AutoML frameworks.
Currently, we are working on new features and trying to improve the performance and the user experience of FEDOT. The major ongoing tasks and plans:
- Effective and ready-to-use pipeline templates for certain tasks and data types;
- Integration with GPU via Rapids framework;
- Alternative optimization methods of fixed-shaped pipelines;
- Integration with MLFlow for import and export of the pipelines;
- Improvement of high-level API.
Also, we are doing several research tasks related to AutoML time-series benchmarking and multi-modal modeling.
Any contribution is welcome. Our R&D team is open for cooperation with other scientific teams as well as with industrial partners.
The general description is available in FEDOT.Docs repository.
Also, a detailed FEDOT API description is available in the Read the Docs.
- The contribution guide is available in the repository.
We acknowledge the contributors for their important impact and the participants of the numerous scientific conferences and workshops for their valuable advice and suggestions.
- The prototype of web-GUI for FEDOT is available in FEDOT.WEB repository.
- Telegram channel for solving problems and answering questions on FEDOT
- Natural System Simulation Team
- Anna Kalyuzhnaya, team leader ([email protected])
- Newsfeed
- Youtube channel
- @article{nikitin2021automated,
- title = {Automated evolutionary approach for the design of composite machine learning pipelines}, author = {Nikolay O. Nikitin and Pavel Vychuzhanin and Mikhail Sarafanov and Iana S. Polonskaia and Ilia Revin and Irina V. Barabanova and Gleb Maximov and Anna V. Kalyuzhnaya and Alexander Boukhanovsky}, journal = {Future Generation Computer Systems}, year = {2021}, issn = {0167-739X}, doi = {https://doi.org/10.1016/j.future.2021.08.022}}
- @inproceedings{polonskaia2021multi,
- title={Multi-Objective Evolutionary Design of Composite Data-Driven Models}, author={Polonskaia, Iana S. and Nikitin, Nikolay O. and Revin, Ilia and Vychuzhanin, Pavel and Kalyuzhnaya, Anna V.}, booktitle={2021 IEEE Congress on Evolutionary Computation (CEC)}, year={2021}, pages={926-933}, doi={10.1109/CEC45853.2021.9504773}}
Other papers - in ResearchGate.