TheilSen5.py

# Compare
plt.figure(figsize = (10, 6))
plt.scatter(
    X,
    y,
    c = 'teal'
)
plt.plot(
    X,
    OLS.predict(X),
    c = 'orange',
    linewidth = 2,
    label = 'OLS Regression'
)
plt.plot(
    X,
    TS.predict(X),
    c = 'magenta',
    linewidth = 2,
    label = 'Theil-Sen Regression'
)
plt.title('Compare regression')
plt.legend(frameon = True, facecolor = 'white')
plt.show()