用朴素贝叶斯完成一个分类问题(最好能给出代码,谢谢各位大佬)

训练数据为train.txt, 测试数据为test.txt, 数据中第一列是类别,后面的是特征
提示:因为特征是连续的,需要将其离散化,一个比较简单方法是计算每列特征的中位数,若大于中位数记为1,否则记为0.