首页 > 解决方案 > 从不同的文件夹加载 sklearn 模型

问题描述

所以我正在尝试用SKlearn建立一个模型。我已经创建了模型并手动将它们放入一个文件夹中。现在我想.py从该特定文件夹加载模型(在不同的文件中)。

我加载模型的代码如下所示。该a变量只是循环不同的模型。它以 1 开头,以 4 结尾

filename = glob.glob('knn_model_folder/knn_model_' + str(a) + '.sav')
loaded_model = joblib.load(open(filename, 'rb'))

我得到的错误:

TypeError: expected str, bytes or os.PathLike object, not list

标签: pythonpython-3.x

解决方案


glob.glob()将返回一个list匹配的路径名。您只需要将路径名作为open函数的字符串:

filename = 'knn_model_folder/knn_model_' + str(a) + '.sav'
loaded_model = joblib.load(open(filename, 'rb'))

推荐阅读