您只需要为该predict
方法提供相同的2D数组,但要提供一个(或多个)要处理的值。简而言之,您可以替换
[0.58,0.76]
用
[[0.58,0.76]]
它应该工作。
编辑:这个答案变得很流行,所以我想我将添加有关ML的更多解释。简短的版本:我们只能predict
在与训练数据(X
)具有相同维数的数据上使用。
在所讨论的示例中,我们给计算机添加了一堆行X
(每个行有2个值),并在中显示了正确的响应y
。当我们想predict
使用新值时,我们的程序期望的是相同的- 行。即使我们只想对一行(具有两个值)进行处理,该行也必须是另一数组的一部分。