I have train a small model and use joblib to save it.
But i get an error while trying to load it.
This is my structure of directory:
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
来添加path:sys.path.append('package/model_folder/model.pkl')
,就像我上面 comments 的那样.但一切都没有改变.
有人能帮我吗?