如何在Matlab中使用具有离散属性的fitcnb（朴素贝叶斯）？

Question

我有一个数据集，如下所示：

第一和第二属性（退款，婚姻状况）是离散属性。第三个属性（应税收入）为连续性属性。

我想在Matlab中建立模型并在此桌子上训练它。

这是我的尝试：

我收到此错误消息：

正态分布不适合类别Yes和预测变量x1的组合。数据具有零方差。

任何人都可以帮助或指导我如何在Matlab中正确使用fitcnb？

Answer 1

我们需要为fitcnb指定类别或离散属性

Mdl = fitcnb（Data，Classes，'CategoricalPredictors'，[1 2]）;