簡體   English   中英

如何讓我的線性回歸 model 使用 clf.predict 進行預測?

[英]how do i get my linear regression model to predict using clf.predict?

因此,我使用線性回歸制作了 model 用於車輛價格預測。 現在我需要讓它預測未來 5 年的價格。 我怎么能用 clf.predict 做到這一點?

X = df[['Year','Engine','FuelType','Age','Transmission','Mileage']]

y = df['Price']這是我的 X 和 y 值

X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2)

clf = LinearRegression()
clf.fit(X_train, y_train)

clf.predict(X_test)   

print(clf.score(X_test, y_test))

在這之后我使用了,

clf.predict([[2022,1300,0,5,0,10000]])

我得到了array([-5722871.63724422])

要僅獲得正值,您應該對 y 值:

y_log = np.log(ytrain.values, where=(ytrain.values>0))

並用模型的預測進行逆變換:

y_pred = exp(clf.predict(X_test))

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM