我正在使用 sklearn 回歸模型來預測每天的銷售額。但是，我希望輸出是一個整數（因為我不能銷售一半的產品），所以我試圖將預測輸出作為一個整數，但我找不到方法。
我知道我可以將 float 轉換為 int，但我的問題是關于調整模型以生成整數預測的可能性（因為輸入也是整數）。
from sklearn.neighbors import KNeighborsRegressor
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import BayesianRidge
model_trained = model.fit(x_train, y_train)
y_est = model_trained.predict(x_test)
print(x_test) --> [645 198 546 619 748]
print(y_est) --> [486.84247399 352.79511474 545.31309247 499.17395514 723.32887889]
uj5u.com熱心網友回復：
predict 方法的回傳值的型別為float。您必須覆寫回歸模型才能更改predict方法的實作。
第 215 行具有以下方法的實作KNeighborsRegressor：
sklearn KNeighborsRegressor 實作
您可以嘗試將第dtype240 行的引數更改為np.intc, 而不是np.float64。以下是可接受的資料型別串列：
numpy 資料型別
我個人認為將 的串列轉換floats為ints.
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/398942.html
標籤：Python 机器学习 scikit-学习 回归 预测