Book Image

Machine Learning with LightGBM and Python

By : Andrich van Wyk
3 (1)
Book Image

Machine Learning with LightGBM and Python

3 (1)
By: Andrich van Wyk

Overview of this book

Machine Learning with LightGBM and Python is a comprehensive guide to learning the basics of machine learning and progressing to building scalable machine learning systems that are ready for release. This book will get you acquainted with the high-performance gradient-boosting LightGBM framework and show you how it can be used to solve various machine-learning problems to produce highly accurate, robust, and predictive solutions. Starting with simple machine learning models in scikit-learn, you’ll explore the intricacies of gradient boosting machines and LightGBM. You’ll be guided through various case studies to better understand the data science processes and learn how to practically apply your skills to real-world problems. As you progress, you’ll elevate your software engineering skills by learning how to build and integrate scalable machine-learning pipelines to process data, train models, and deploy them to serve secure APIs using Python tools such as FastAPI. By the end of this book, you’ll be well equipped to use various -of-the-art tools that will help you build production-ready systems, including FLAML for AutoML, PostgresML for operating ML pipelines using Postgres, high-performance distributed training and serving via Dask, and creating and running models in the Cloud with AWS Sagemaker.
Table of Contents (17 chapters)
1
Part 1: Gradient Boosting and LightGBM Fundamentals
6
Part 2: Practical Machine Learning with LightGBM
10
Part 3: Production-ready Machine Learning with LightGBM

GPU training for LightGBM

The LightGBM library has native support for training the model on a GPU [1]. Two GPU platforms are supported: GPU via OpenCL and CUDA. Leveraging the GPU via OpenCL offers support for the broadest range of GPUs (including AMD GPUs) and is significantly faster than running the model on a CPU. However, the CUDA platform offers the fastest runtime if you have an NVIDIA GPU available.

Setting up LightGBM for the GPU

Setting up your environment to use the GPU can be a bit tricky, but we'll review the core steps here.

Note

The GPU setup steps discussed here are offered as a guide and overview of the process of setting up your environment. The exact version number of libraries and drivers listed here may be outdated, and it’s recommended that you review the official documentation for up-to-date versions: https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html.

In order to use the GPU, we have to compile and build the LightGBM library...