You are on page 1of 2

Códigos da Regressão linear:

 import matplotlib.pyplot as plt


 from sklearn.datasets import load_diabetes
 from sklearn.linear_model import LinearRegression
 from sklearn.metrics import mean_squared_error
 import numpy as np

 du_dy = np.array([40,60,90,100,150,200,400,800])
 tau = np.array([30,120,140,170,230,380,430,870])

 lm_model = LinearRegression()
 lm_model.fit(du_dy.reshape(-1,1), tau)
 slope = lm_model.coef_
 intercept = lm_model.intercept_

 print("b0: \t{}".format(intercept))
 print("b1: \t{}".format(slope[0]))

 plt.plot(du_dy, tau, 'p', ls=':',label='dados experimentais')
 plt.title('Regressão linear\n')
 plt.xlabel('Taxa de deformação (Y [1/s])')
 plt.ylabel('Tensão Cisalhante (Tau [Pa])')
 plt.scatter(du_dy, tau, s=30)
 plt.plot(du_dy, (du_dy* slope + intercept), color='purple',label=
'regressão linear')
 plt.legend()
 plt.grid()
 plt.show(

You might also like