joblib：无法从父目录的另一个子文件夹加载转储的模型

Question

我训练了一个小模型并使用 joblib 来保存它。
但我在尝试加载时遇到错误。
这是我的目录结构：

loader.py
package
    model_folder
        __init__.py
        model.py
        model.pkl
    predict_folder
        __init__.py
        predict.py

在

package.predict_folder.predict.py

内部，我编写了一个函数来加载

model.pkl

。

from ..model. import model
import joblib
import sys

def predict(model_path):
    # sys.path.append(model_path)
    model = joblib.load(model_path)

然后在

loader.py

中，我调用该函数。

from package.predict_folder.predict import predict
predict('package/model_folder/model.pkl')

但我收到此错误：

ModuleNotFoundError: No module named 'model_folder'

我尝试使用

sys

添加路径：

sys.path.append('package/model_folder/model.pkl')

就像我上面评论的那样。但一切都没有改变。 谁能帮助我吗？

Answer 1

您在 Predict.py 中的导入语句中似乎有一个拼写错误。

更改此行：

来自..模型。进口型号

至：

从..model导入模型

这应该可以修复 ModuleNotFoundError。