데이터 세트가 제공되었으므로 일부 X 및 Y 데이터 사이의 관계를 찾으려고합니다. sklearn 라이브러리를 사용하여 데이터를 플롯하고 eqn의 곡선을 예측/예측할 수 있기를 원합니다.Sklearn to and Polynomial Regression을 사용하여 곡선의 방정식을 맞추거나 예측하십시오. 무한 루프 오류
그러나 내 데이터 집합에 다항식 회귀 모델을 적용한 후 예상 값을 플로팅하려 할 때 코드가 무한 루프에 멈추었습니다.
최종 목표는 일단 곡선을 예측/계획하면 곡선의 전체 방정식이 무엇인지 알 수 있기를 원합니다.
여기 내 코드가 있습니다.
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
Dataset = pd.DataFrame()
Dataset["X"] = [6377, 6378, 6379, 6380, 6381, 6382, 6383, 6385, 6387, 6392, 6397, 6402]
Dataset["Y"] = [1.225, 1.112, 1.007, 0.9093, 0.8194, 0.7364, 0.6601, 0.5258, 0.4135, 0.1948, 0.08891, 0.04008]
print(Dataset)
X = np.reshape(np.array(Dataset['X']), (1, -1))
Y = np.reshape(np.array(Dataset['Y']), (1, -1))
print(X)
print(Y)
from sklearn.linear_model import LinearRegression
linReg = LinearRegression()
linReg.fit(X, Y)
plt.scatter(X, Y, color='red')
# plt.plot(X,linReg.predict(X), color = 'blue')
from sklearn.preprocessing import PolynomialFeatures
polyREG = PolynomialFeatures(degree=4)
xPoly = polyREG.fit_transform(X)
LinReg2 = LinearRegression()
LinReg2.fit(xPoly, Y)
#
# try:
# xgrid = np.arange(min(X), max(X), .1)
# except Exception as e:
# print(e)
# xgrid = range(6377, 6403, 1)
# xgrid = np.asarray(xgrid)
# print(xgrid.shape)
# xgrid = np.reshape(xgrid, (1,-1))
xgrid = np.reshape(np.arange(6300, 6405, 1), (1,-1))
print(xgrid.shape)
#X = np.reshape(np.array(Dataset['X']), (1, -1))
#plt.plot(xg, 1, color = "blue")
try:
plt.plot(xgrid, LinReg2.predict(polyREG.fit_transform(xgrid)), color='blue')
except Exception as e:
print(e)
plt.show()
예상되는 곡선을 그리기 위해 어떻게 수정해야합니까? – NoviceCoder