You are on page 1of 1

import matplotlib.

pyplot as plt
import numpy as np
import pandas as pd

dataset = pd.read_csv('p9.csv')

x = dataset.iloc[:,0] #it'll read first column


domain=x
y = dataset.iloc[:,1] #it'll read second column

def local_reg(x0,x,y,tau):
x0= [1,x0]
x=[[1,i]for i in x]
x=np.asarray(x)
xw=(x.T)*np.exp(np.sum((x-x0)**2,axis=1)/(-2*(tau**2)))
beta=(np.linalg.pinv(xw @ x)) @ (xw @ y)
return (beta @ x0)

def draw(tau):
pred=[local_reg(x0, x, y, tau)for x0 in domain]
plt.plot(x,y,'o',color='green')
plt.plot(x,pred,'o',color='red',linewidth=5)
plt.show

draw(4) #you can chane this number

You might also like