TPOT is built on top of several existing Python libraries, including:
Most of the necessary Python packages can be installed via the Anaconda Python distribution, which we strongly recommend that you use. We also strongly recommend that you use of Python 3 over Python 2 if you're given the choice.
You can install TPOT using pip
or conda-forge
.
pip
NumPy, SciPy, scikit-learn, pandas, joblib, and PyTorch can be installed in Anaconda via the command:
conda install numpy scipy scikit-learn pandas joblib pytorch
DEAP, tqdm and stopit can be installed with pip
via the command:
pip install deap tqdm stopit
Optionally, you can install XGBoost if you would like TPOT to use the eXtreme Gradient Boosting models. XGBoost is entirely optional, and TPOT will still function normally without XGBoost if you do not have it installed. Windows users: pip installation may not work on some Windows environments, and it may cause unexpected errors.
pip install xgboost
If you have issues installing XGBoost, check the XGBoost installation documentation.
If you plan to use Dask for parallel training, make sure to install dask[delay] and dask[dataframe] and dask_ml.
pip install dask[delayed] dask[dataframe] dask-ml fsspec>=0.3.3
If you plan to use the TPOT-MDR configuration, make sure to install scikit-mdr and scikit-rebate:
pip install scikit-mdr skrebate
To enable support for PyTorch-based neural networks (TPOT-NN), you will need to install PyTorch. TPOT-NN will work with either CPU or GPU PyTorch, but we strongly recommend using a GPU version, if possible, as CPU PyTorch models tend to train very slowly.
We recommend following PyTorch's installation instructions customized for your operating system and Python distribution.
Finally to install TPOT itself, run the following command:
pip install tpot
conda-forge
To install tpot and its core dependencies you can use:
conda install -c conda-forge tpot
To install additional dependencies you can use:
conda install -c conda-forge tpot xgboost dask dask-ml scikit-mdr skrebate
As mentioned above, we recommend following PyTorch's installation instructions for installing it to enable support for PyTorch-based neural networks (TPOT-NN).
Installation problems
Please file a new issue if you run into installation problems.