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

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)

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

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

Oleg Kostromin

Iryna Kondrashchenko

Marco Pasini