Skip to content

dfavenfre/Bike-Sharing-Demand-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Bike Rent Demand Prediction Model

Within the project, it is purposed to build a demand forecast model for bike sharing count for Capital Bikeshare program in Washington, D.C.. The data is available at Kaggle (https://www.kaggle.com/competitions/bike-sharing-demand/data).

We've written a class that enables multi-model building, as well as an optimization script with Optuna for RMSE (root mean squared error). As an example for the whole project, a master notebook is also available within the repository.

Data Pre-processing

First, We increased the number of features by dummifying categorical variables, however, since the data we have is quite small in terms of size, modelling it this way would've dropped the models' score significantly.

As a consideration for the linear model, transforming categorical variables into numeric ones by using either get_dummies of Pandas or Onehot Encoding of Sklearn are more meaningful and thus they provide better score when compared to not transforming them into numeric.

However, in tree-based models like Decision-Tree, XGBoost, LGBM, RandomForest, etc., not applying these methodologies boosts the tree-based models to understand nonlinear relationships between features and the target, and thus, the models yield better and significant results.

As stated earlier, the data include way too much categorical features to apply any encoding algorithm (or non-linear relationships might not be caught by tree algorithms) therefore, some of these features, which are under the certain thresholds specified for feature engineering, may be over/underestimated when it comes to feature importance.

Consequently, we decided to move forward without encoding for categorical variables within the data pre-process. The remaining data was way too perfect, with no missing values or outliers.

Variable Description

1

Preliminary Inference (Correlation Heatmap)

image

Example Usage of BuildRegressionModel.py

# pre-define hyperparameters
hyperparameters_lgb = {
    
    "n_estimators":[200,2000],
    "learning_rate":[0.01,0.10],
    "num_leaves":[20,1000],
    "max_depth":[3,12],
    "min_data_in_leaf":[200,10000],
    "max_bin":[200,300],
    "lambda_l1":[0,80],
    "lambda_l2":[0,100],
    "min_gain_to_split":[0,30]
}

# instantiate models
brm = BuildRegressionModel(X,y) # feature and target should be assigned earlier

# shuffle and split data
X_train, X_test, y_train, y_test = brm.shuffle_data() # if no parameter is passed, the defaults are; Shuffle=True, test_size=0.3

# build the optimized XGB model with Optuna
study_xgb = optuna.create_study(direction='minimize',study_name='XGBregression')
study_xgb.optimize(BuildRegressionModel.xgb_obj, n_trials=100) # the optimization is stopped after 100 trials 

model_xgb = xgboost.XGBRegressor(**study_xgb.best_params)
model_xgb.fit(X_train, y_train)
y_pred_xgb = model_xgb.predict(X_test)

# Optimized XGBoosting Model Results
brm.get_regression_result(y_test, y_pred_xgb)

Optuna Optimization Details

def xgb_obj(trial, hyperparameters=hyperparameters_xgb):

    """
    Description:
    -----------
        Optimize Extreme Gradient Boosting with pre-defined hyperparameters. The tuning is optimized for minimizing the error terms, which is rooted means squared errors (RMSE).  

    Parameters:
    -----------
        Hyperparameters(dict): Pre-defined dictionary consist of key and value pairs of hyperparameters,
    
    Returns:
    -----------
        Process (Verbose) lines with RMSE scores and hyperparameters used in each trial. 
    
    """

    param = {}

    for key, value in hyperparameters.items():
        if isinstance(value, Iterable):
            if isinstance(value[0], float):
                param[key] = trial.suggest_float(key, value[0], value[1])
            else:
                param[key] = trial.suggest_int(key, value[0], value[1])
        else:
            param[key] = value

    # non-hyperparameter settings
    param["n_jobs"] = -1 # deploy 100% of gpu's computational power 
    param["random_state"] = 42

    model = xgboost.XGBRegressor(**param)
    model.fit(X_train, y_train)
    y_pred = model.predict(X_test)

    return (mean_squared_error(y_test, y_pred))**(1/2)

Hyperparameter fine-tuning with Optuna

Optuna is an automatic hyperparameter optimization software framework, particularly designed for machine learning. We used Optuna both for finding best parameters and score and for visualizing the importance and relations to conclude which of the hyperparameters add value the most to the model. Slice plots provide a better perspective when it comes to fine-tuning the model. Slice plot shows, given the hyperparameter upper and lower bounds, which hyperparameter yields better results and the plot also shows a boundary for us to decide whether to increase or decrease upper boundary. Thus, it scopes us for more improvement.

Extreme Gradient Boosting Model Regression Hyperparameter Importances

newplot (1)

newplot

Light Gradient Boosting Model Regression Hyperparameter Importances

newplot (2)

newplot (3)

Controlling for futures' individual contribution using SHAP

SHAP (Shapley Additive Explanations) is a game theoretic approach to explain the output of any machine learning model. It connects optimal credit allocation with local explanations using the classic Shapley values from game theory and their related extensions. It measures each inputs' individual contribution to the model. To exemplify, let's say we have X, Y, Z as features within the model. First the algorithm builds a model using X, Y, and Z and then it sequentially drops each feature and builds another model to obtain amongst which of these features add the most value to the model. Hence, the following SHAP visualizations follow the exact procedure. We visualized both LGBM and XGBM SHAP outputs to observe which feature adds more weight in terms of significance to the model.

XGB Model Output

image

LGB Model Output

image

Ensemble Regression Model

Ensemble models combine the decisions from multiple models to improve the overall performance. This can be achieved in various ways. A voting ensemble involves making a prediction that is the average of multiple other regression models. Stacking is also an ensemble learning technique that uses predictions from multiple models (for example decision tree, knn or svm) to build a new model.

Conclusion

We've applied several Machine Learning Regression models, including XGBM, LGBM, Stacked and Bagging Ensemble algorithms, and the below table summarizes our findings. Depending on the goal, minimizing rmse or maximixing r-square, the below table should be able to help you out.

Model R-square score RMSE score
Stacked Ensemble 0.975 35.341
Bagging Ensemble 0.960 45.465
Optimized XGB 0.969 32.553
Optimized LGB 0.970 39.209