Skip to content

Instantly share code, notes, and snippets.

@Alrecenk
Created September 8, 2013 07:42
Show Gist options
  • Save Alrecenk/6482703 to your computer and use it in GitHub Desktop.
Save Alrecenk/6482703 to your computer and use it in GitHub Desktop.
Calculate the output for a Naive Bayes classifier.
//Calculate the probability that the given input is in the positive class
public double probability(double in[]){
double relativepositive=0,relativenegative=0;
for(int j=0; j<in.length; j++){
relativepositive += (in[j]-posmean[j])*(in[j]-posmean[j]) / posvariance[j] ;
relativenegative += (in[j]-negmean[j])*(in[j]-negmean[j]) / negvariance[j] ;
}
relativepositive = positives*Math.exp(0.5*relativepositive) ;
relativenegative = negatives*Math.exp(0.5*relativenegative) ;
return relativepositive / (relativepositive + relativenegative) ;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment