机器学习Joblib预测

Question

我从这个repo github repo中得到了一些保存的模型，我正在尝试从这些模型中做出预测，我是机器学习的新手。这就是我拥有的。

这就是我正在使用的notebook笔记本

import joblib
from sklearn.feature_extraction.text import CountVectorizer
import numpy as np
from sklearn.preprocessing import OneHotEncoder
import pandas as pd

model_mnb = joblib.load("/content/decision_tree.joblib")
model_dec = joblib.load("/content/gradient_boost.joblib")
model_gra = joblib.load("/content/mnb.joblib")
model_forest = joblib.load("/content/random_forest.joblib")

print(model_mnb, model_dec, model_gra, model_forest)

sample = ["I have Fever", "I have headache", "I have sore limbs"]

encoder = OneHotEncoder(sparse=False)
test_data = encoder.fit_transform(sample)
model_mnb.predict(test_data)

我得到了：

ERROR:  Expected 2D array, got 1D array instead:

Answer 1

这很可能是因为你的样本变量是一个样本。如果您要预测多个样本，您将需要一个嵌套列表，这可能是predict()函数所期望的。尝试在test_data变量或更早的sample变量中添加一对方括号。