클래스 기능 활용하기

이제 Document 클래스의 __init__ 메서드에 텍스트를 자동으로 전처리하는 기능을 추가했어요. 이번 연습에서는 실제 사용자 입장에서 여러분의 작업이 어떤 이점을 주는지 확인해 보겠습니다.

아래에 복사된 Document 클래스가 여러분의 환경에 로드되어 있어요(방금 추가한 업데이트 포함).

class Document:
  def __init__(self, text):
    self.text = text
    # pre tokenize the document with non-public tokenize method
    self.tokens = self._tokenize()
    # pre tokenize the document with non-public count_words
    self.word_counts = self._count_words()

  def _tokenize(self):
    return tokenize(self.text)

  # non-public method to tally document's word counts with Counter
  def _count_words(self):
    return Counter(self.tokens)