Как исправить регрессию параболы, чтобы график регрессии был гладким? - PullRequest
0 голосов
/ 17 марта 2020
import math
import random
from matplotlib import pyplot as plt
from sklearn.linear_model import LinearRegression
import numpy as np
%matplotlib inline
x = np.array([[i] for i in range(-5 , 5)])
x1 = x ** 2
y = [7 , 5 , 4 , 3 , 1 , 0 , -1  ,2 , 5 , 6]
lr = LinearRegression().fit(x1 , y)
plt.scatter(x , y)
predictions = lr.predict(x1)
plt.plot(x , predictions)
plt.show()

Списки x и y могут быть изменены, но главная проблема в том, что график не является гладким, и я также не получаю часть предсказания, кто-нибудь может помочь?

...