This book uses (mostly) Python 3.7 and various ML- and trading-related libraries available in three different conda environments based on the Miniconda distribution. I developed the content on Ubuntu 20.04 while also testing on Mac OS 10.15 (Catalina).
Update: Release 2.0 reduces the number of environments to 2 and bumps the Python version to 3.8 for the main
ml4t
and to 3.6 for thebacktest
environment. Instructions below reflect these changes.
To update the Docker image to the latest version, run:
docker pull appliedai/packt:latest
Depending on your OS, you may have several options to create these environments. These are, in increasing order of complexity:
- Recommended: use Docker Desktop to pull an image from Docker Hub and create a local container with the requisite software to run the notebooks.
- Create the conda environments using the provided
.yml
environment files as outlined below. However, the backtesting environment that relies on the patched version ofzipline
only exists for Ubuntu due to numerous version conflicts. - If you are experienced (and work on a UNIX-based system), you can also create your own virtual environments and install the libraries required for the different notebooks using
pip
as needed.
We'll describe the first two options in turn. Then, we address how to work with Jupyter notebooks to view and execute the code examples. Finally, we list additional installation instructions for libraries that require non-python dependencies like TA-Lib for technical analysis.
Docker Desktop is a very popular application for MacOS and Windows machines because is permits for the easy sharing of containerized applications across different OS. For this book, we have a Docker image that let's you instantiate a container to run Ubuntu 20.04 as a guest OS with the pre-installed conda environments on Windows 10 or Mac OS X without worrying about dependencies on your host.
As usual, installation differs for Mac OS X and Window 10, and requires an additional step for Windows 10 Home to enable virtualization.
We'll cover installation for each OS separately and then address some setting adjustments necessary in both cases.
Installing Docker Desktop on Mac OS X is very straightforward:
- Follow the detailed guide in Docker docs to download and install Docker Desktop from Docker Hub. It also covers how Docker Desktop and Docker Toolbox can coexist.
- Use homebrew by following the tutorial here.
Open terminal and run the following test to check that Docker works:
docker run hello-world
Review the Getting Started guide for Mac OS to familiarize yourself with key settings and commands.
Docker Desktop works on both Windows 10 Home and Pro editions; the Home edition requires the additional step of enabling the Virtual Machine Platform.
You can now install Docker Desktop on Windows Home machines using the Windows Subsystem for Linux (WSL 2) backend. Docker Desktop on Windows Home is a full version of Docker Desktop for Linux container development.
Windows 10 Home machines must meet certain requirements. These include Windows 10 Home version 2004 (released May 2020) or higher. The Docker Desktop Edge release also supports Windows 10, version 1903 or higher.
Enable WSL 2 as described here, taking the following steps:
- Enable the optional Windows Subsystem for Linux feature. Open PowerShell as Administrator and run:
dism.exe /online /enable-feature /featurename:Microsoft-Windows-Subsystem-Linux /all /norestart
- Check that your system meets the requirements outlined here and update your Windows 10 version if necessary.
- Enable the Virtual Machine Platform optional feature by opening PowerShell as and Administrator and run:
dism.exe /online /enable-feature /featurename:VirtualMachinePlatform /all /norestart
- Restart your machine to complete the WSL install and update to WSL 2.
- Download and run the Linux kernel update package. You will be prompted for elevated permissions, select ‘yes’ to approve this installation.
- Set WSL 2 as your default version when installing a new Linux distribution by open PowerShell as Administrator and run the following command:
wsl --set-default-version 2
Once we have enabled WSL 2 for Windows Home, the remaining steps to install Docker Desktop are the same for Windows 10 Home and Pro, Enterprise or Education. Refer to the linked guides for each OS version for system requirements.
- Download and run (double-click) the installer from Docker Hub.
- When prompted, ensure the Enable Hyper-V Windows Features option is selected on the Configuration page.
- Follow the instructions on the installation wizard to authorize the installer and proceed with the install.
- When the installation is successful, click Close to complete the installation process.
- If your admin account is different to your user account, you must add the user to the docker-users group. Run Computer Management as an administrator and navigate to Local Users and Groups > Groups > docker-users. Right-click to add the user to the group. Log out and log back in for the changes to take effect.
Open Powershell and run the following test to check that Docker works:
docker run hello-world
Review the Getting Started guide for Windows to familiarize yourself with key settings and commands.
The getting started guides for each OS referenced above describe the Docker Desktop settings.
- Under Preferences, look for Resources to find out how you can increase the memory allocated to the container; the default setting is too low given the size of the data. Increase to at least 4GB, better 8GB or more.
- Several examples are quite memory-intensive, for example the NASDAQ tick data and the SEC filings example in Chapter 2, and will require significantly higher memory allocation.
We will download the code examples and data to the local drive on your host OS but run it from the Docker container by mounting your local drive as a volume. This should work fine with the current versions but in case you receive permission errors , please refer to the File Sharing sections in the Docker user guides. The Docker GUIs let you assign permissions explicitly. See also (slightly outdated) explanation here.
You can work with the code samples by downloading a compressed version of the GitHub repository, or by cloning its content. The latter will result in a larger download because it includes the commit history.
Alternatively, you can create a fork of the repo and continue to develop from there after cloning its content.
To work with the code locally, do the following:
- Select a file system location where you would like to store the code and the data.
- Using the
ssh
orhttps
links or the download option provided by the greenCode
button on the GitHub repository, either clone or unzip the code to the target folder.- To clone the starter repo, run
git clone https://github.com/stefan-jansen/machine-learning-for-trading.git
and change into the new directory. - If you cloned the repo and did not rename it, the root directory will be called
machine-learning-for-trading
, the ZIP the version will unzip tomachine-learning-for-trading-master
.
- To clone the starter repo, run
To download US equity data that we'll be using for several examples throughout the book in the next step, register for a personal Quandl account to obtain an API key. It will be displayed on your profile page.
If you are on a UNIX-based system like Mac OSX, you may want to store the API key in an environment variable such as QUANDL_API_KEY, e.g. by adding export QUANDL_API_KEY=<your_key>
to your .bash_profile
.
We'll be using a Docker image based on the Ubuntu 20.04 OS with Anaconda's miniconda Python distribution installed. It comes with two conda environments described below.
With a single Docker command, we can accomplish several things at once (see the Getting Started guides linked above for more detail):
- only on the first run: pull the Docker image from the Docker Hub account
appliedai
and the repositorypackt
with the taglatest
- creates a local container with the name
ml4t
and runs it in interactive mode, forwarding the port 8888 used by thejupyter
server - mount the current directory containing the starter project files as a volume in the directory
/home/packt/ml4t
inside the container - set the environment variable
QUANDL_API_KEY
with the value of your key (that you need to fill in for<your API key>
), and - start a
bash
terminal inside the container, resulting in a new command prompt for the userpackt
.
- Open a Terminal or a Powershell window.
- Navigate to the directory containing the ML4T code samples that you sourced above.
- In the root directory of the local version of the repo, run the following command, taking into account the different path formats required by Mac and Windows:
-
Mac OS: you can use the
pwd
command as a shell variable that contains the absolute path to the present working directory (and you could use$QUANDL_API_KEY
if you created such an environment variable in the previous step):docker run -it -v $(pwd):/home/packt/ml4t -p 8888:8888 -e QUANDL_API_KEY=<your API key> --name ml4t appliedai/packt:latest bash
-
Windows: enter the absolute path to the current directory with forward slashes, e.g.
C:/Users/stefan/Documents/machine-learning-for-trading
instead ofC:\Users\stefan\Documents\machine-learning-for-trading
, so that the command becomes (for this example):docker run -it -v C:/Users/stefan/Documents/machine-learning-for-trading:/home/packt/ml4t -p 8888:8888 -e QUANDL_API_KEY=<your API key> --name ml4t appliedai/packt:latest bash
-
- Run
exit
from the container shell to exit and stop the container. - To resume working, you can run
docker start -a -i ml4t
from Mac OS terminal or Windows Powershell in the root directory to restart the container and attach it to the host shell in interactive mode (see Docker docs for more detail).
To update the Docker image to the latest version, run:
docker pull appliedai/packt:latest
Now you are running a shell inside the container and can access both conda environments. Run conda env list
to see that there are a base
, ml4t
(default), and a backtest
environments.
The backtest
environment is necessary because the latest version of Zipline 1.4.1 only support Python 3.6 and older versions of various other dependencies that partly also require compilation. I hope to update Zipline in the future to run on Python 3.8 as well.
We use the environment ml4t
except for a dozen notebooks related to backtesting that use Zipline directly inputs generated by Zipline. The noteooks that require the backtest
environment contain a notification.
If you want to use a GPU for the deep learning examples, you can run
conda install tensorflow-gpu
if you have the proper CUDA version installed. Alternatively, you can leverage TensorFlow's Docker images and install any additional libraries there; the DL examples don't require anything that's overly complicated to install.
- You can switch to another environment using
conda activate <env_name>
or using the Jupyter Notebook or Jupyter Lab Kernel menu thanks to the nb_conda_kernels extension (see below). - You may see an error message suggesting you run
conda init bash
. After doing so, reload the shell with the commandsource .bashrc
.
To run Zipline backtests, we need to ingest
data. See the Beginner Tutorial for more information.
The image has been configured to store the data in a .zipline
directory in the directory where you started the container (which should be the root folder of the starter code you've downloaded above).
From the command prompt of the container shell, run
conda activate backtest
zipline ingest
You should see numerous messages as Zipline processes around 3,000 stock price series.
I have patched the following country code issue in the latest Zipline version, so you should not have to manually fiddle with the asset database any longer.
When running a backtest, you will likely encounter an error because the current Zipline version requires a country code entry in the exchanges table of the assets-7.sqlite
database where it stores the asset metadata.
The linked GitHub issue describes how to address this by opening the SQLite database and entering US
in the country_code
field of the exchanges table.
In practice, this looks as follows:
- Use the SQLite Browser to open the file
assets-7.sqlite
in the directory containing your latest bundle download. The path will look like this (on Linux/Max OSX) if you ran the command as just described:~/machine-learning-for-trading/data/.zipline/data/quandl/2020-12-29T02;06;08.894865/
- Select the table
exchanges
as outlined in the following screenshot:
That's all. Unfortunately, you (had to..) repeat this everytime you run zipline ingest
.
You can run juypter notebooks using either the traditional notebook or the more recent Jupyter Lab interface; both are available in all conda
environments. Moreover, you start jupyter from the base
environment and switch the environment from the notebook due to the nb_conda_kernels
package (see docs.
To get started, run one of the following two commands:
jupyter notebook --ip 0.0.0.0 --no-browser --allow-root
jupyter lab --ip 0.0.0.0 --no-browser --allow-root
There are also alias
shortcuts for each so you don't have to type them:
nb
for thejupyter notebook
version, andlab
for thejupyter lab
version.
The container terminal will display a few messages while spinning up the jupyter server. When complete, it will display a URL that you should paste into your browser to access the jupyter server from the current working directory.
You can modify any of the environments using the standard conda workflow outlined below; see Docker docs for how to persist containers after making changes.
The code examples use Anaconda's miniconda distribution to facilitate dependency management.
If you are experienced (and work in a Unix-based environment), feel free to create your own environment using the provided .yml
files (however, these are not tested, do not cntain pinned version numbers and should be viewed as a list of directly required libraries without their dependencies (which conda
will handle).
You could also just install the packages required for the notebooks you are interested in; the most recent versions (as of March 2021) should work. However, you may find it rather challenging to get the patched Zipline version to work properly.
The Docker image is intended as the default way of running the code as the notebooks are only tested against the environments it contains. Unfortunately, I'm not able to provide support for debugging your own environments.
The notebooks rely on two different virtual environments based on miniconda3 that you need to install first.
You can find detailed instructions for various operating systems here.
[conda] is the package manager provided by the Anaconda python distribution that is tailored to facilitating the installation of data science libraries.
Just like for virtual environments for generic python installations, conda
permits the creation of separate environments that are based on the same interpreter (miniconda3 if you followed the above instructions) but can contain different package and versions of packages. See also here for a more detailed tutorial.
You can create a new conda environment with name env_name
and one or more packages with a specific version number using the command:
conda create --name env_name package=version_number
e.g.
conda create --name pandas_environment pandas=1.05
Here, we will create the ml4t
environment from file to ensure you install the library versions that the code has been tested with. To create the environment with the name ml4t
(tested on Ubuntu) for linux
, from the repository's root directory, just run:
conda env create -f installation/linux/ml4t.yml
or, for Max OS X
conda env create -f installation/macos/ml4t.yml
from the command line in this directory.
After you've create it, you can activate the environment using its name, which in our case is ml4t
:
conda activate ml4t
To deactivate, simply use
conda deactivate
This section covers how to set up notebook extension that facilitate working in this environment and how to convert notebooks to python script if preferred.
jupyter notebooks can use a range of extentsion provided by the community. There are many useful ones that are described in the documentation.
The notebooks in this repo are formatted to use the Table of Contents (2) extension. For the best experience, activate it using the Configurator in the Nbextensions tab available in your browser after starting the jupyter server. Modify the settings to check the option 'Leave h1 items out of ToC' if not set by default.
The book uses jupyter notebooks to present the code with extensive commentary and context information and facilitate the visualization of results in one place. Some of the code examples are longer and make more sense to run as python
scripts; you can convert a notebook to a script by running the following on the command line:
$ jupyter nbconvert --to script [YOUR_NOTEBOOK].ipynb
For the python wrapper around TA-Lib, please follow installation instructions here.