Installation | FLAML (original) (raw)

Python

FLAML requires Python version >= 3.7. It can be installed from pip:

or conda:

conda install flaml -c conda-forge

Optional Dependencies

Autogen

pip install "flaml[autogen]"

Task-oriented AutoML

pip install "flaml[automl]"

• openai models

pip install "flaml[openai]"

• catboost

pip install "flaml[catboost]"

• vowpal wabbit
• time series forecaster: prophet, statsmodels

pip install "flaml[forecast]"

• huggingface transformers

Notebook

To run the notebook examples, install flaml with the [notebook] option:

pip install "flaml[notebook]"

Distributed tuning

• ray
• spark

Spark support is added in v1.1.0

pip install "flaml[spark]>=1.1.0"

For cloud platforms such as Azure Synapse, Spark clusters are provided. But you may also need to install Spark manually when setting up your own environment. For latest Ubuntu system, you can install Spark 3.3.0 standalone version with below script. For more details of installing Spark, please refer to Spark Doc.

sudo apt-get update && sudo apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
    ca-certificates-java ca-certificates openjdk-17-jdk-headless \
    && sudo apt-get clean && sudo rm -rf /var/lib/apt/lists/*
wget --progress=dot:giga "https://www.apache.org/dyn/closer.lua/spark/spark-3.3.0/spark-3.3.0-bin-hadoop2.tgz?action=download" \
    -O - | tar -xzC /tmp; archive=$(basename "spark-3.3.0/spark-3.3.0-bin-hadoop2.tgz") \
    bash -c "sudo mv -v /tmp/\${archive/%.tgz/} /spark"
export SPARK_HOME=/spark
export PYTHONPATH=/spark/python/lib/py4j-0.10.9.5-src.zip:/spark/python
export PATH=$PATH:$SPARK_HOME/bin

• nni
• blendsearch

pip install "flaml[blendsearch]"

• synapse

To install flaml in Azure Synapse and similar cloud platform

pip install flaml[synapse]

Test and Benchmark

• test
• benchmark

pip install flaml[benchmark]

.NET

FLAML has a .NET implementation in ML.NET, an open-source, cross-platform machine learning framework for .NET.

You can use FLAML in .NET in the following ways:

Low-code

• Model Builder - A Visual Studio extension for training ML models using FLAML. For more information on how to install, see the install Model Builder guide.
• ML.NET CLI - A dotnet CLI tool for training machine learning models using FLAML on Windows, MacOS, and Linux. For more information on how to install the ML.NET CLI, see the install the ML.NET CLI guide.

Code-first

• Microsoft.ML.AutoML - NuGet package that provides direct access to the FLAML AutoML APIs that power low-code solutions like Model Builder and the ML.NET CLI. For more information on installing NuGet packages, see the install and use a NuGet package in Visual Studio or dotnet CLI guides.

To get started with the ML.NET API and AutoML, see the csharp-notebooks.