def adbPredictor(df):
dataTrainX, dataTrainY, dataTestX, dataTestY = sample(df)
# clf = linear_model.SGDRegressor()
clf = ensemble.AdaBoostRegressor()
clf.fit(dataTrainX, dataTrainY)
predicted = clf.predict(dataTestX)
fig, ax = plotter.subplots()
ax.set_ylabel('Predicted KNN Weekly')
ax.scatter(dataTestY, predicted)
ax.set_xlabel('Measured')
predicted = np.reshape(predicted, (predicted.size, 1))
corrCoeff = pearsonr(dataTestY,predicted)
print(corrCoeff[0])
plotter.show()
return predicted
评论列表
文章目录