MultinomialNB、GaussianNB或CategoricalNB在这里使用什么？

Question

让我有一个输入特性X = {X1, X2}。其中X1是实值的(也可以认为它遵循高斯分布)，但X2是一个分类特征。现在如果我想使用朴素贝叶斯算法。我应该用哪一个？GaussianNB在分类特征方面的另一种完美表现吗？

Answer 1

使用get_dummies() (pandas库)转换你的类别特征X2。然后训练模型。

我建议先试试GaussianNB，然后评估他的准确性。然后尝试sklearn拥有的其他朴素贝叶斯模型。在没有看到数据的情况下(即使有数据)，很难预测哪种模型在每种情况下工作得更好。对每一个进行评估。