<p align="center"> <picture> <img alt="Falcon logo" src="https://gist.githubusercontent.com/OKUA1/55e2fb9dd55673ec05281e0247de6202/raw/41063fcd620d9091662fc6473f9331a7651b4465/falcon.svg" height = "250"> </picture> </p>

FALCON: A Lightweight AutoML Library

Falcon is a lightweight python library that allows to train production-ready machine learning models in a single line of code.

Why Falcon ? 🔍

• Simplicity: With Falcon, training a comprehensive Machine Learning pipeline is as easy as writing a single line of code.
• Flexibility: Falcon offers a range of pre-set configurations, enabling swift interchangeability of internal components with just a minor parameter change.
• Extendability: Falcon's modular design, along with its extension registration procedure, allows seamless integration with virtually any framework.
• Portability: A standout feature of Falcon is its deep native support for ONNX models. This lets you export complex pipelines into a single ONNX graph, irrespective of the underlying frameworks. As a result, your model can be conveniently deployed on any platform or with almost any programming language, all without dependence on the training environment.

Future Developments 🔮

Falcon ML is under active development. We've already implemented a robust and production-ready core functionality, but there's much more to come. We plan to introduce many new features by the end of the year, so stay tuned!

⭐ If you liked the project, please support us with a star!

Quick Start 🚀

You can try falcon out simply by pointing it to the location of your dataset.

from falcon import AutoML

AutoML(task = 'tabular_classification', train_data = '/path/to/titanic.csv')

Alternatively, you can use one of the available demo datasets.

from falcon import AutoML
from falcon.datasets import load_churn_dataset, load_insurance_dataset 
# churn -> classification; insurance -> regression

df = load_churn_dataset()

AutoML(task = 'tabular_classification', train_data = df)

Installation 💾

Stable release from PyPi

pip install falcon-ml

Latest version from GitHub

pip install git+https://github.com/OKUA1/falcon

Installing some of the dependencies on Apple Silicon Macs might not work, the workaround is to create an X86 environment using Conda

conda create -n falcon_env
conda activate falcon_env
conda config --env --set subdir osx-64
conda install python=3.9
pip3 install falcon-ml

Documentation 📚

You can find a more detailed guide as well as an API reference in our official docs.

Authors & Contributors ✨

<table> <tbody> <tr> <td align="center"><a href="https://www.linkedin.com/in/oleh-kostromin-b671a4157/"><img src="https://avatars.githubusercontent.com/u/48349467?v=4" width="100px;" alt=""/><br />_{<b>Oleg Kostromin</b>}</a><br /></td> <td align="center"><a href="https://www.linkedin.com/in/iryna-kondrashchenko-673800155/"><img src="https://avatars.githubusercontent.com/u/72279145?v=4" width="100px;" alt=""/><br />_{<b>Iryna Kondrashchenko</b>}</a><br /></td> <td align="center"><a href="https://www.linkedin.com/in/pasinimarco/"><img src="https://avatars.githubusercontent.com/u/50598094?v=4" width="100px;" alt=""/><br />_{<b>Marco Pasini</b>}</a><br /></td> </tr> </tbody> </table>