[英]How do I add a linear regression line to this scatter plot?
如何將 ML model 生成的線性回歸線添加到散點圖 plot 中?
pickle_in=open("student-model.pickle","rb")
linear=pickle.load(pickle_in)
acc=linear.score(x_test, y_test)
print(f"accuracy= {round(acc*100,2)}%")
#comment: for scatter plot
style.use("ggplot")
p="G1"
pyplot.scatter(data[p],data["G3"])
pyplot.xlabel(p)
pyplot.ylabel("Final Grade")
pyplot.show()
你可以試試:
from numpy.polynomial.polynomial import polyfit
b, m = polyfit(x, y, 1)
pyplot.plot(x, b + m * x, '-')
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.