diff --git a/classifier.py b/classifier.py index 7e3cbe76..04a6f6a9 100644 --- a/classifier.py +++ b/classifier.py @@ -28,8 +28,9 @@ def __init__(self, documents: Dict[str, List[str]]): for word, frequency in self._get_word_freqs(doc).items(): self._word_freqs[current_class][word] += frequency self._vocabulary.add(word) - #TODO: Calculate self._logprior and self._loglikelihood for all classes - # and words. Don't forget about smoothing. + raise NotImplementedError("Calculate self._logprior and " + "self._loglikelihood for all classes and " + "words. Don't forget about smoothing.") # TODO def classify(self, document: str): """