python-3.x - 在 Python3 中通过 Pickle 加载模型时出错

Question

我正在尝试在 Python3 中创建一个对象的实例，并使用 pickle 从另一个 module.py 文件导入，其代码如下：

上课缺勤模型（）：

def __init__(self, model_file, scaler_file):
    # read the 'model' and 'scaler' files which were saved
    with open('model','rb') as model_file, open('scaler', 'rb') as scaler_file:
        
        self.reg = pickle.load(model_file)
        self.scaler = pickle.load(scaler_file)
        self.data = None
    

以及创建类实例时的错误

model = absenteeism_model('model','scaler')

显示为：

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-4-cb3169ea39c9> in <module>
----> 1 model = absenteeism_model('model','scaler')

~\ML_Goals\Absenteeism_from_work_exercise\Integration\absenteeism_module.py in __init__(self, model_file, scaler_file)
     45             with open('model','rb') as model_file, open('scaler', 'rb') as scaler_file:
     46 
---> 47                 self.reg = pickle.load(model_file)
     48                 self.scaler = pickle.load(scaler_file)
     49                 self.data = None

ModuleNotFoundError: No module named 'sklearn.linear_model.logistic'

我在 module.py 文件中导入了以下模块：

import numpy as np
import pandas as pd
import pickle

from sklearn.linear_model import LogisticRegression
from sklearn import  metrics

请提供帮助，我将不胜感激任何解决它的建议。

标签： python-3.xjupyter-notebookpickle