2000 character limit reached
Naive Bayes with Correlation Factor for Text Classification Problem (1905.06115v1)
Published 8 May 2019 in cs.IR, cs.LG, and stat.ML
Abstract: Naive Bayes estimator is widely used in text classification problems. However, it doesn't perform well with small-size training dataset. We propose a new method based on Naive Bayes estimator to solve this problem. A correlation factor is introduced to incorporate the correlation among different classes. Experimental results show that our estimator achieves a better accuracy compared with traditional Naive Bayes in real world data.