I want to build a model that describes a curve that fits the data shown in the scatterplot. I thought it would be straight forward using sklearn. But the choice and application of the different methods gets rather confusing.

Which algorithms would you use to tackle this problem?

enter image description here

  • 1
    $\begingroup$ Welcome to our site. Could you explain the sense in which you use "vectors" in your title? This appears to be a (standard) univariate regression problem, not a vector problem. $\endgroup$ – whuber Sep 4 '18 at 13:40
  • $\begingroup$ Vector in a computational sense. c(1,2,3,4,5) or [1,2,3,4,5] or array(1,2,3,4,5) $\endgroup$ – Benni Sep 4 '18 at 19:58
  • $\begingroup$ The only way that would apply here would be to represent the $x$ coordinates as one vector and the $y$ coordinates as another. But what, then, could you possibly mean by a "nonlinear" relationship between the two of them? $\endgroup$ – whuber Sep 4 '18 at 20:35

I suppose the answer depends on what you are trying to accomplish:

  • Predicting future values
  • Performing statistical inference
  • Other ?

My answer is that you can use any of the many regression models available and choose the one that you believe to be the most appropriate using whichever metric you are comfortable with. Here are a few examples along with the Python Sklearn code

Polynomial linear regression

import numpy as np
import matplotlib.pyplot as plt
from sklearn.preprocessing import PolynomialFeatures
from sklearn.linear_model import LinearRegression
from sklearn.pipeline import Pipeline, make_pipeline

model_1 = make_pipeline(PolynomialFeatures(degree = 5),LinearRegression())
plt.figure(figsize = (8,5))
plt.scatter(x,y, alpha = .3, label = 'Data')
plt.plot(x,model_1.predict(x.reshape(-1,1)), color = 'red', label = 'Model')
plt.title('Polynomial degree 5')
plt.xlabel('X'), plt.ylabel('Y')
plt.legend(), plt.show()

enter image description here

Decision tree regression

from sklearn.tree import DecisionTreeRegressor
model_2 = DecisionTreeRegressor(max_depth = 3)

enter image description here

Piecewise linear spline interpolation

from scipy import interpolate
tck = interpolate.splrep(x, y, k=1, s=1, t = [35])
plt.plot(x,interpolate.splev(x, tck, der=0), color = 'red', label = 'Model')

enter image description here

  • $\begingroup$ Thank you for your clear answer. I'm certainly searching for a smooth curve in my model to predict y from all possible x. The polynomial regression is the first step. $\endgroup$ – Benni Sep 4 '18 at 20:00
  • $\begingroup$ You can also try LOESS regression and cubic splines if you are looking for smooth models - wont be able to do as much inference on those though $\endgroup$ – Xavier Bourret Sicotte Sep 4 '18 at 21:51

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.