Calculate noisy channel probability

1 files changed, 11 insertions, 0 deletions

diff --git a/spellchecker/src/ConfusionMatrixReader.java b/spellchecker/src/ConfusionMatrixReader.java
index c9e79ab..90a8cef 100644
--- a/spellchecker/src/ConfusionMatrixReader.java
+++ b/spellchecker/src/ConfusionMatrixReader.java
@@ -60,4 +60,15 @@ public class ConfusionMatrixReader {
         Integer count = confusionMatrix.get(error + "|" + correct);
         return count == null ? 0 : count;
     }
+
+    /**
+     * Given the error {@code error}, find the number of occurrences of the
+     * error.
+     *
+     * @param error
+     * @return
+     */
+    public int getErrorsCount(String error) {
+        return countMatrix.getOrDefault(error, 0);
+    }
 }