Machine Learning

Key Concepts

Introduction

Machine learning is a field of study that combines statistics and computer science to build intelligent systems that predict outcomes. Quant researchers commonly use machine learning models to optimize portfolios, make trading signals, and manage risk. These models can find relationships in datasets that humans struggle to find, are subtle, or are too complex. You can use machine learning techniques in your research notebooks.

Supported Libraries

The following table shows the supported machine learning libraries:

LibraryResearch TutorialDocumentation
KerasTutorialDocumentation
TensorFlowTutorialDocumentation
Scikit-LearnTutorialDocumentation
hmmlearnTutorialDocumentation
gplearnTutorialDocumentation
PyTorchTutorialDocumentation
Stable BaselinesTutorialDocumentation
tslearnTutorialDocumentation
XGBoostTutorialDocumentation

Add New Libraries

To request a new library, contact us. We will add the library to the queue for review and deployment. Since the libraries run on our servers, we need to ensure they are secure and won't cause harm. The process of adding new libraries takes 2-4 weeks to complete. View the list of libraries currently under review on the Issues list of the Lean GitHub repository.

Transfer Models

You can load machine learning models from the Object Store or a custom data file like pickle. If you train a model in the Research Environment, you can also save it into the Object Store to transfer it to the backtesting and live trading environment.

You can also see our Videos. You can also get in touch with us via Discord.

Did you find this page helpful?

Contribute to the documentation: