In [5]:
# Iz sklearn biblioteke uvozimo funkcije za rad sa linearnom regresijom
from sklearn.linear_model import LinearRegression
In [6]:
# Pravimo model koji ćemo koristiti za izračunavanje parametara linearne regresije
model = LinearRegression(fit_intercept=True)
In [7]:
# Slično kao i kod LDA algoritma koristimo naredbu fit
model.fit(x[:, np.newaxis], y)
Out[7]:
LinearRegression(copy_X=True, fit_intercept=True, n_jobs=1, normalize=False)