Python polynomial2_features Exemples

Exemple #1

import datasets
import regression

X, Y = datasets.load_nonlinear_example1()
ex_X = datasets.polynomial2_features(X)
print(ex_X)

model = regression.LinearRegression()
model.fit(ex_X, Y)
print(model.theta)
print(model.predict(ex_X))
print(model.score(ex_X, Y))

Exemple #2

import numpy as np
import datasets
import regression

X, Y = datasets.load_nonlinear_example1()
ex_X = datasets.polynomial2_features(X)
model = regression.LinearRegression()
model.fit(ex_X, Y)

samples = np.arange(0, 4, 0.1)
x_samples = np.c_[np.ones(len(samples)), samples]
ex_x_samples = datasets.polynomial2_features(x_samples)

import matplotlib.pyplot as plt
plt.scatter(X[:, 1], Y)
plt.plot(samples, model.predict(ex_x_samples))
plt.show()