标签: python numpy pandas scikit-learn
使用GaussianNB()时,如何修复不一致的样本数?另外,输入pandas dataframe是否可以作为model.fit函数的参数?
答案 0 :(得分:2)
问题在于GaussianNB期望weather处于(n_samples, n_features)形状。您目前将其作为一维数组,因此GaussianNB将其解释为具有14个要素的1个样本。
GaussianNB
weather
(n_samples, n_features)
要转换为正确的形状,您可以使用this答案中所述的weather[:,None]。所以,以下应该可以做到这一点:
weather[:,None]
model.fit(weather[:,None], play)