Polynomial regression can be very useful. There isn’t always a linear relationship between X and Y. Sometime the relation is exponential or Nth order.
Related course: Machine Learning Intro for Python Developers
You can plot a polynomial relationship between X and Y. If there isn’t a linear relationship, you may need a polynomial. Unlike a linear relationship, a polynomial can fit the data better.
You create this polynomial line with just one line of code.
poly_fit = np.poly1d(np.polyfit(X,Y, 2))
That would train the algorithm and use a 2nd degree polynomial.
After training, you can predict a value by calling polyfit, with a new example. It will then output a continous value.
The example below plots a polynomial line on top of the collected data. It trains the algorithm, then it makes a prediction of a continous value.
import numpy as np
It’s important to not overfit or underfit, you want to capture the relationship but not follow the points exactly. A linear relationship would underfit, overfitting would be picking the degree so high that it fits the points. Instead, you want to capture the relationship.