from sklearn.datasets import load_boston import numpy as np import pandas as pd data=pd.read_csv('./data_picture/chapter1/boston_house_prices.csv') data.head() from sklearn.model_selection import train_test_split X=data.drop('MEDV',axis=1) y=data['MEDV'] X=X.values y=y.values X_train,X_test,y_train,y_test=train_test_split(X,y,random_state=33,test_size=0.25) from sklearn.preprocessing import StandardScaler ss_X=StandardScaler() scaler_X=ss_X.fit(X_train) X_train=scaler_X.transform(X_train) X_test=scaler_X.transform(X_test) from sklearn.neural_network import MLPRegressor model = MLPRegressor(solver='lbfgs', hidden_layer_sizes=(15,15), random_state=1) model.fit(X_train, y_train) from sklearn.metrics import r2_score,mean_squared_error,mean_absolute_error print('训练集回归评估指标:') model_score1=model.score(X_train,y_train) print('The accuracy of train data is',model_score1) print('测试集回归评估指标:') model_score2=model.score(X_test,y_test) print('The accuracy of test data is',model_score2) y_test_predict=model.predict(X_test) mse=mean_squared_error(y_test,y_test_predict) print('The value of mean_squared_error:',mse) mae=mean_absolute_error(y_test,y_test_predict) print('The value of mean_absolute_error:',mae)解析一下这段代码
时间: 2023-12-24 21:27:01 浏览: 419
这段代码是一个机器学习模型的训练和评估过程,具体步骤如下:
1. 从sklearn.datasets模块中导入boston数据集。
2. 使用pandas模块读取boston房价数据集,将其存储在data变量中,并显示前5行数据。
3. 从sklearn.model_selection模块中导入train_test_split函数,将数据集划分为训练集和测试集,其中训练集占75%,测试集占25%。
4. 从sklearn.preprocessing模块中导入StandardScaler类,用于对数据进行标准化处理。
5. 对训练集和测试集中的特征数据X进行标准化处理,使用fit()函数对scaler_X进行训练,再使用transform()函数对X_train和X_test进行标准化处理。
6. 从sklearn.neural_network模块中导入MLPRegressor类,用于创建多层感知机回归模型。
7. 创建MLPRegressor对象,并指定模型参数:solver='lbfgs', hidden_layer_sizes=(15, 15), random_state=1。其中,solver参数指定优化算法为拟牛顿法L-BFGS,hidden_layer_sizes参数指定模型中隐藏层的节点数为15,random_state参数指定随机种子。
8. 使用fit()函数对模型进行训练,输入训练集的特征数据X_train和目标数据y_train。
9. 从sklearn.metrics模块中导入r2_score、mean_squared_error和mean_absolute_error函数,用于评估模型的性能。
10. 分别使用score()函数计算训练集和测试集的R2得分,并输出结果。
11. 使用predict()函数对测试集进行预测,计算预测结果与实际结果之间的均方误差和平均绝对误差,并输出结果。
阅读全文