机器学习：模型评估和模型保存

一、模型评估

from sklearn.metrics import accuracy_score, confusion_matrix, classification_report# 使用测试集进行预测y_pred = model.predict(X_test)# 计算准确率accuracy = accuracy_score(y_test, y_pred)print(f"Accuracy: {accuracy*100:.2f}%")# 打印混淆矩阵conf_matrix = confusion_matrix(y_test, y_pred)print("Confusion Matrix:")print(conf_matrix)# 打印分类报告，包括精确率、召回率和F1分数class_report = classification_report(y_test, y_pred)print("Classification Report:")print(class_report)

二、模型保存

#使用joblib保存模型import joblibjoblib.dump(model, "./yorelee_model.pth")#模型的后缀名是无所谓的

三、后话

���型选择的时候，也可以使用模型融合，即结果由用不同模型的结果按比例得到。

比如pre=(pre_1*a+pre_2*b)/(a+b)。

那么我们在保存模型的时候，这两个模型要一起保存，然后之后训练就导入两个模型，pre这样算出来就行。

%%time# 2种模型融合def model_mix(pred_1, pred_2):    result = pd.DataFrame(columns=['LinearRegression','XGBRegressor','Combine'])    for a in range (80):        for b in range(1,80):                    y_pred = (a*pred_1 + b*pred_2 ) / (a+b)                                        mse = mean_squared_error(y_test,y_pred)                                        mse = mean_squared_error(y_test,y_pred)                    new_row = pd.DataFrame([{'LinearRegression':a,                                              'XGBRegressor':b,                                              'Combine':mse}])                    result = pd.concat([result, new_row], ignore_index=True)    return resultlinear_predict=model_linear.predict(x_test)xgb_predict=XGBClassifier.predict(x_test)model_combine = model_mix(linear_predict,  xgb_predict)model_combine.sort_values(by='Combine', inplace=True)model_combine.head()#各种比例来一份，看看mse最高分，查看 a和b的具体值