使用GaussianNB()时,如何修复不一致的样本数?

时间:2016-03-11 19:07:17

标签: python numpy pandas scikit-learn

使用GaussianNB()时,如何修复不一致的样本数?另外,输入pandas dataframe是否可以作为model.fit函数的参数?

enter image description here enter image description here

1 个答案:

答案 0 :(得分:2)

问题在于GaussianNB期望weather处于(n_samples, n_features)形状。您目前将其作为一维数组,因此GaussianNB将其解释为具有14个要素的1个样本。

要转换为正确的形状,您可以使用this答案中所述的weather[:,None]。所以,以下应该可以做到这一点:

model.fit(weather[:,None], play)