How to optimise learning rates in XGBoost example 2?

How to optimise learning rates in XGBoost example 2?

How to optimise learning rates in XGBoost example 2?

This recipe helps you optimise learning rates in XGBoost example 2


Recipe Objective

Many a times while working on a dataset and using a Machine Learning model we don"t know which set of hyperparameters will give us the best result. Passing all sets of hyperparameters manually through the model and checking the result might be a hectic work and may not be possible to do.

To get the best set of hyperparameters we can use Grid Search. Grid Search passes all combinations of hyperparameters one by one into the model and check the result. Finally it gives us the set of hyperparemeters which gives the best result after passing in the model.

So this recipe is a short example of how can optimise multiple parameters in XGBoost.

Step 1 - Import the library - GridSearchCv

from sklearn import datasets from sklearn.model_selection import train_test_split from xgboost import XGBClassifier from sklearn.model_selection import GridSearchCV from sklearn.model_selection import StratifiedKFold import numpy

Here we have imported various modules like numpy, test_train_split, datasets, XGBClassifier and GridSearchCV from differnt libraries. We will understand the use of these later while using it in the in the code snipet.
For now just have a look on these imports.

Step 2 - Setup the Data

Here we have used datasets to load the inbuilt wine dataset and we have created objects X and y to store the data and the target value respectively. dataset = datasets.load_wine() X = y = X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25)

Step 3 - Using Model

Here, we are using XGBClassifier as a Machine Learning model to use GridSearchCV. So we have created an object model. model = XGBClassifier()

Step 5 - Parameters to be optimized

In XGBClassifier we want to optimise learning rate by GridSearchCV. So we have set the parameter as a list of values form which GridSearchCV will select the best value of parameter. n_estimators = [100, 200, 300, 400, 500] learning_rate = [0.0001, 0.001, 0.01, 0.1] param_grid = dict(learning_rate=learning_rate, n_estimators=n_estimators) kfold = StratifiedKFold(n_splits=10, shuffle=True, random_state=7)

Step 6 - Using GridSearchCV and Printing Results

Before using GridSearchCV, lets have a look on the important parameters.

  • estimator: In this we have to pass the models or functions on which we want to use GridSearchCV
  • param_grid: Dictionary or list of parameters of models or function in which GridSearchCV have to select the best.
  • Scoring: It is used as a evaluating metric for the model performance to decide the best hyperparameters, if not especified then it uses estimator score.
  • cv: It signifies the number of splits in cross validation.
Making an object grid_search for GridSearchCV and fitting the dataset i.e X and y grid_search = GridSearchCV(model, param_grid, scoring="neg_log_loss", n_jobs=-1, cv=kfold) grid_result =, y) Now we are using print statements to print the results. It will give the values of hyperparameter as a result. print("Best: %f using %s" % (grid_result.best_score_, grid_result.best_params_)) means = grid_result.cv_results_["mean_test_score"] stds = grid_result.cv_results_["std_test_score"] params = grid_result.cv_results_["params"] for mean, stdev, param in zip(means, stds, params): print("%f (%f) with: %r" % (mean, stdev, param)) As an output we get:

Best: -0.077744 using {"learning_rate": 0.1, "n_estimators": 200}
-1.086577 (0.000539) with: {"learning_rate": 0.0001, "n_estimators": 100}
-1.074744 (0.001073) with: {"learning_rate": 0.0001, "n_estimators": 200}
-1.063105 (0.001605) with: {"learning_rate": 0.0001, "n_estimators": 300}
-1.051657 (0.002128) with: {"learning_rate": 0.0001, "n_estimators": 400}
-1.040397 (0.002643) with: {"learning_rate": 0.0001, "n_estimators": 500}
-0.986714 (0.005129) with: {"learning_rate": 0.001, "n_estimators": 100}
-0.891275 (0.009533) with: {"learning_rate": 0.001, "n_estimators": 200}
-0.808661 (0.013511) with: {"learning_rate": 0.001, "n_estimators": 300}
-0.736632 (0.016334) with: {"learning_rate": 0.001, "n_estimators": 400}
-0.673480 (0.018467) with: {"learning_rate": 0.001, "n_estimators": 500}
-0.443098 (0.032668) with: {"learning_rate": 0.01, "n_estimators": 100}
-0.236960 (0.048792) with: {"learning_rate": 0.01, "n_estimators": 200}
-0.159834 (0.052822) with: {"learning_rate": 0.01, "n_estimators": 300}
-0.125159 (0.056987) with: {"learning_rate": 0.01, "n_estimators": 400}
-0.108292 (0.059112) with: {"learning_rate": 0.01, "n_estimators": 500}
-0.083225 (0.059937) with: {"learning_rate": 0.1, "n_estimators": 100}
-0.077744 (0.057482) with: {"learning_rate": 0.1, "n_estimators": 200}
-0.077754 (0.057472) with: {"learning_rate": 0.1, "n_estimators": 300}
-0.077754 (0.057472) with: {"learning_rate": 0.1, "n_estimators": 400}
-0.077754 (0.057472) with: {"learning_rate": 0.1, "n_estimators": 500}

Relevant Projects

PySpark Tutorial - Learn to use Apache Spark with Python
PySpark Project-Get a handle on using Python with Spark through this hands-on data processing spark python tutorial.

Machine Learning or Predictive Models in IoT - Energy Prediction Use Case
In this machine learning and IoT project, we are going to test out the experimental data using various predictive models and train the models and break the energy usage.

Data Science Project-TalkingData AdTracking Fraud Detection
Machine Learning Project in R-Detect fraudulent click traffic for mobile app ads using R data science programming language.

Customer Churn Prediction Analysis using Ensemble Techniques
In this machine learning churn project, we implement a churn prediction model in python using ensemble techniques.

Predict Credit Default | Give Me Some Credit Kaggle
In this data science project, you will predict borrowers chance of defaulting on credit loans by building a credit score prediction model.

Predict Census Income using Deep Learning Models
In this project, we are going to work on Deep Learning using H2O to predict Census income.

Data Science Project - Instacart Market Basket Analysis
Data Science Project - Build a recommendation engine which will predict the products to be purchased by an Instacart consumer again.

Ensemble Machine Learning Project - All State Insurance Claims Severity Prediction
In this ensemble machine learning project, we will predict what kind of claims an insurance company will get. This is implemented in python using ensemble machine learning algorithms.

Predict Churn for a Telecom company using Logistic Regression
Machine Learning Project in R- Predict the customer churn of telecom sector and find out the key drivers that lead to churn. Learn how the logistic regression model using R can be used to identify the customer churn in telecom dataset.

Choosing the right Time Series Forecasting Methods
There are different time series forecasting methods to forecast stock price, demand etc. In this machine learning project, you will learn to determine which forecasting method to be used when and how to apply with time series forecasting example.