Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- Double probability = 0.0;
- probability += Math.log((classCounts.get(currentClass)+ alpha)/(numOfTrainingExamples + alpha * classes.size())); //prior
- subMap = classFeatureCounts.get(currentClass);
- for(String token : docTokens){
- int classTokenCount = subMap.containsKey(token)?subMap.get(token):0;
- int classAllTokensCount = subMap.get("*");
- probability += Math.log((classTokenCount + alpha)/(classAllTokensCount + alpha * vocabularySize));
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement