python - 使用 model.predict() 时出现错误“将字符串转换为浮点数”

Question

当我使用model.predict()时，出现以下错误，

ValueError：无法将字符串转换为浮点数：

我的代码中使用了One-hot 编码，如下所示，

features_df = pd.get_dummies(df, columns=['garage_type', 'city'])

使用的预测代码如下所示。

sample = [["attached"], ["Richardport"]]
predictedvalue = model.predict(sample)'

有什么建议可以解决这个错误吗？谢谢你。

标签： pythonpandasdataframescikit-learn

Answer 1

您收到错误的原因是因为您正在向模型发送字符串，而模型期望浮点向量与您的训练数据具有相同的形状。

从代码中可以清楚地看出，在使用数据进行训练之前，您正在使用一种热编码。您还需要将样本转换为相同的一种热编码矢量格式才能得到答案。