使用Pickle加载随机森林模型rb文件时遇EOFError报错求助

问题详情 #

使用pickle加载RF_Model_Py3模型文件时，出现以下报错：

Exception has occurred: EOFError
Ran out of input
  File "C:\Users\HP\Downloads\Crop-Yield-Prediction-using-ML-master (1)\Crop-Yield-Prediction-using-ML-master\RF_predict.py", line 15, in <module>
    RanFor = pickle.load(f)

对应的加载代码：

import pickle
from sklearn.ensemble import RandomForestRegressor
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import classification_report,accuracy_score,roc_auc_score

with open('RF_Model_Py3', 'rb') as f:
    RanFor = pickle.load(f)

排查与解决步骤 #

• 检查模型文件状态
  查看RF_Model_Py3的文件大小，若为0字节，说明模型保存过程未完成，需重新训练并保存。保存时务必使用wb模式且确保代码执行完毕：

  # 完整保存模型示例
  import pickle
  from sklearn.ensemble import RandomForestRegressor
  
  # 假设已完成模型训练
  rf_model = RandomForestRegressor()
  rf_model.fit(X_train, y_train)
  
  with open('RF_Model_Py3', 'wb') as f:
      pickle.dump(rf_model, f)
  

• 确认文件路径准确性
  若使用相对路径，确保模型文件与RF_predict.py在同一目录；或直接使用绝对路径定位文件：

  with open('C:/Users/HP/Downloads/Crop-Yield-Prediction-using-ML-master (1)/Crop-Yield-Prediction-using-ML-master/RF_Model_Py3', 'rb') as f:
      RanFor = pickle.load(f)
  

• 统一Python版本
  pickle跨版本兼容性较差，确保保存模型和加载模型使用的Python版本一致（如均为3.8及以上）。

• 验证文件完整性
  若模型文件是下载所得，可能因中断导致不完整，重新下载或重新生成模型文件。

• 换用pickle.loads加载
  先读取完整文件内容再加载，排查读取环节问题：

  with open('RF_Model_Py3', 'rb') as f:
      data = f.read()
      if not data:
          print("模型文件为空")
      else:
          RanFor = pickle.loads(data)
  

内容的提问来源于stack exchange，提问作者Kathirvel Rajmohan