sklearn 如何使用多个保存的模型合并和预测数据

Question

我想用我训练过的 4 个模型来预测我的数据。所以我试图将我的模型合并到一个列表中，但是在我附加我的模型之后，我不能调用“预测”并且我得到了这样的错误：

AttributeError: 'list' object has no attribute 'predict'

我的代码是这样的：

vect_tes = features.transform(frame['text'])


model = [[]]
for i in range(4):
    mod = open('model_'+str(i+1)+'.pkl', 'rb')
    model.append(pickle.load(mod))
    mod.close()

predict = model.predict(vect_tes)

Answer 1

不确定合并模型，我认为最好查看ensemble methods。

关于AttributeError，您最初将模型声明为列表列表： model = [[]]。因此，当您将经过训练的模型附加到循环中时，模型会存储[ [], model1, model2, model3, model4 ]，因此如果您想使用您的模型，则需要访问该列表，例如：model[1].predict。您还可以在初始时将模型声明为 []，在这种情况下，您将从 0 索引而不是 1 访问您的模型。