import pandas as pd import numpy as np import matplotlib.pyplot as plt from sklearn import metrics #定义评估回归性能函数 def evaluation_metrics() : print("Mean Absolute Error:",metrics.mean_absolute_error(y_test,y_test_pred)) print("Mean Squared Error:",metrics.mean_squared_error(y_test,y_test_pred)) print("Root Mean Squared Error:",np.sqrt(metrics.mean_squared_error(y_test,y_test_pred))) print("r2_score : ",str(metrics.r2_score(y_test,y_test_pred))) #定义绘制Expermental-Predicted图像的函数 def graph(model) : plt.grid(True) plt.plot([1,12],[1,12],color='darkgrey',lw=3) plt.scatter(y_train,y_train_pred,color='c') plt.scatter(y_test,y_test_pred,color='salmon') plt.xlabel("Expermental Values",fontsize=14,fontweight='bold') plt.ylabel("Predicted Values",fontsize=14,fontweight='bold') plt.xlim(1,12) plt.ylim(1,12) plt.title(str(model),fontweight='bold') plt.savefig(str(model)+ "_test.tif",dpi=500) plt.show()

这段代码定义了两个函数，分别是evaluation_metrics和graph。

evaluation_metrics函数用于评估回归性能，其中使用了sklearn库中的mean_absolute_error、mean_squared_error、r2_score等函数来计算预测值与实际值之间的差距，并输出结果。

graph函数用于绘制Expermental-Predicted图像，其中使用了matplotlib库来绘制散点图和参考直线。该函数接受一个参数model，用于指定模型的名称，绘制完成后将图像保存为tif格式并显示出来。