Как добавить линию линейной регрессии на этот график рассеяния?

Question

Как добавить линию линейной регрессии, сгенерированную моделью ML, на график рассеяния?

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()

Answer 1

Вы можете попробовать:

from numpy.polynomial.polynomial import polyfit

b, m = polyfit(x, y, 1)
pyplot.plot(x, b + m * x, '-')