我需要一个用于大规模朴素贝叶斯的库,能够处理数百万的训练样本和超过10万的二进制特征。必须是可在线更新的版本(训练后可更新)。我还需要top-k输出,即对单个实例进行多重分类。准确性不是很重要。
目的是用于自动文本分类应用。
任何关于优秀库的建议都非常受欢迎。
编辑:库最好是Java语言的。
回答:
如果除了朴素贝叶斯之外的其他学习算法也可以接受,那么可以查看Vowpal Wabbit(C++),它被认为是最好的可扩展文本分类算法之一(在线随机梯度下降 + LDA)。我不确定它是否支持top-K输出。