diff --git a/code_prediction.py b/code_prediction.py index 41728d5..7cc6847 100644 --- a/code_prediction.py +++ b/code_prediction.py @@ -32,9 +32,9 @@ def prediction(input,output): temp_skeptic_logprob += skeptic_class_logprob + word_logprobs['skeptic'][token] if temp_paranormal_logprob > temp_skeptic_logprob: - output_file.write('0\n') - else: output_file.write('1\n') + else: + output_file.write('0\n') ## def main(): prediction('dev-0/in.tsv','dev-0/out.tsv')