TheilSen3.py

# OLS
OLS = LinearRegression()
OLS.fit(X, y.ravel())

# Plot
plt.figure(figsize = (10, 6))
plt.scatter(
    X,
    y,
    c = 'teal'
)
plt.plot(
    X,
    OLS.predict(X),
    c = 'orange',
    linewidth = 2,
)
plt.title('OLS Regression')
plt.show()