数据分析——简单的线性方程

it2024-01-30  71

from sklearn.linear_model import LinearRegression import pandas as pd import numpy as np import matplotlib.pyplot as plt data = pd.read_excel('./car_data.xls') # 创建模型,将模型在当前的数据场景中进行训练,生成真正的模型 lr = LinearRegression() X = data.loc[:,'TV Ads'] y = data.loc[:,'Cars Sold'] X = (X.values).reshap(-1,1) lr.fit(X,y) k = lr.coef_[0] print(k) b = lr.intercept_ print(b) y_predict = lr.predict(X) print(y_predict) print(plt.scatter(X,y)) print(plt.plot(X,y_predict)) 创建模型实例将数据进行特征工程的处理(1-D转换成2-D)将特征工程处理过后的数据,进行模型的训练利用生成的模型进行预测
最新回复(0)