Output config + naive bayes
Fix naive bayes