Deal with Infinity score in test (#133)

jekyll · Jan 17, 2017 · 795bed1 · 795bed1
1 parent be47188
commit 795bed1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/bayes/bayesian_integration_test.rb b/test/bayes/bayesian_integration_test.rb
@@ -57,7 +57,7 @@ def classification_scores(classifier)
     @testing_set.collect do |line|
       parts = line.strip.split("\t")
       result, score = classifier.classify_with_score(parts.last)
-      "#{result}:#{score}"
+      score.infinite? ? "irrelevant" : "#{result}:#{score}"
     end
   end
 end