Machine Learning with C++ – Polynomial Regression on GPU

Machine Learning with C++ – Polynomial Regression on GPU


Hello, this is my second article about how to use modern C++ for solving machine learning problems. This time I will show how to make a model for polynomial regression problem described in previous article, but now with another library which allows you to use your GPU easily.

For this tutorial I chose MShadow library, you can find documentation for it here. This library was chosen because it is actively developed now, and used as a basis for one of a wide used deep learning framework MXNet. Also it is a header only library with minimal dependencies, so it’s integration is not hard at all.
Continue reading the article and source code here. Please feel free to leave comment or create issue in repository if you find some mistakes.
Content

Preparations
Loading data to MShadow data-structures
Standardization
Generating additional polynomial components
Generating new data for testing model predictions
Batch gradient descent implementation
Training the regression model
Making predictions
Plot results



Link: Machine Learning with C++ – Polynomial Regression on GPU