bayes_kotlin_6.kt

fun classifyText( text : String ) : Int {
    val class_1_probability = findProbabilityGivenSample( text , positiveBagOfWords )
    val class_2_probability = findProbabilityGivenSample( text , negativeBagOfWords )
    if ( class_1_probability > class_2_probability ) {
        return CLASS_POSITIVE
    }
    else if ( class_1_probability < class_2_probability ) {
        return CLASS_NEGATIVE
    }
}