/Python/Piecewise linear regression python

Piecewise linear regression python

from scipy import optimize

def piecewise_linear(x, x0, x1, b, k1, k2, k3):
    condlist = [x < x0, (x >= x0) & (x < x1), x >= x1]
    funclist = [lambda x: k1*x + b, lambda x: k1*x + b + k2*(x-x0), lambda x: k1*x + b + k2*(x-x0) + k3*(x - x1)]
    return np.piecewise(x, condlist, funclist)

p , e = optimize.curve_fit(piecewise_linear, x, y)

xd = np.linspace(-30, 30, 1000)
plt.plot(x, y, "o")
plt.plot(xd, piecewise_linear(xd, *p))

Source: songhuiming.github.io

Similar codes

12 votes180 views5 answers
Django-cors-headers
Python
4 votes180 views5 answers
How to take multiple integer input in python
Python
1 vote158 views2 answers
Python pandas shape
Python
1 vote158 views1 answer
Python read string between two substrings
Python
17 votes224 views6 answers
Odd or even in python
Python
92 votes179 views6 answers
Python choose random element from list
Python
0 vote139 views1 answer
Select Children of an Object With ForeignKey in Django?
Python
0 vote161 views1 answer
Restore tf model python ValueError: Unknown loss function:smoothL1
Python
0 vote157 views1 answer
Python If Boolean example
Python
0 vote127 views1 answer
Python empty constructor
Python