/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

0 vote279 views1 answer
Get random line from file python
Python
0 vote262 views1 answer
How to repeat if statement in python
Python
0 vote271 views1 answer
Multiplication of two or more numbers in python
Python
0 vote284 views1 answer
Tf MaxPooling2D
Python
0 vote263 views1 answer
Countvectorizer remove stop words
Python
0 vote312 views1 answer
Use ipython magic in script
Python
0 vote292 views1 answer
App = Flask(_name_) NameError: name '_name_' is not defined
Python
0 vote276 views2 answers
How to capitalize first letter in python in list using list comprehension
Python
25 votes293 views10 answers
Github
Python
39 votes272 views4 answers
Round down number python
Python