标签:mat test train LinearRegression import 线性 np model sklearn
import numpy as np import sklearn.datasets #加载原数据 from sklearn.model_selection import train_test_split #分割数据 from matplotlib import pyplot as plt from sklearn.linear_model import LinearRegression #创建数据 def createdata(): boston = sklearn.datasets.load_boston() databoston = boston.data m,n = np.shape(databoston) one_mat = np.ones((m,1)) databoston = np.column_stack((databoston,one_mat)) lableboston = boston.target x_train,x_test,y_train,y_test= train_test_split(databoston,lableboston,test_size=0.2) #分割数据测试数据为30% x_train = np.mat(x_train) y_train = np.mat(y_train).reshape(-1,1) x_test = np.mat(x_test) y_test = np.mat(y_test).reshape(-1,1) # print(x_train[1,:],len(x_train)) # print(y_train[1],len(y_train)) return x_train,x_test,y_train,y_test x_train, x_test, y_train, y_test = createdata() model = LinearRegression(copy_X=True, fit_intercept=False, n_jobs=1, normalize=False) model.fit(x_train,y_train) print('系数矩阵:\n',model.coef_) print('线性回归模型:\n',model) # 使用模型预测 predicted = model.predict(x_test) axix_x1 = np.linspace(0,2*len(y_test),len(y_test)) plt.plot(axix_x1, y_test,'b-') plt.plot(axix_x1, predicted,'r--') # 显示图形 plt.show()
标签:mat,test,train,LinearRegression,import,线性,np,model,sklearn 来源: https://www.cnblogs.com/cxhzy/p/10764490.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。