__init__.py 501 B

123456789101112131415161718
  1. from __future__ import absolute_import
  2. from .tfidf import TFIDF
  3. from .textrank import TextRank
  4. try:
  5. from .analyzer import ChineseAnalyzer
  6. except ImportError:
  7. pass
  8. default_tfidf = TFIDF()
  9. default_textrank = TextRank()
  10. extract_tags = tfidf = default_tfidf.extract_tags
  11. set_idf_path = default_tfidf.set_idf_path
  12. textrank = default_textrank.extract_tags
  13. def set_stop_words(stop_words_path):
  14. default_tfidf.set_stop_words(stop_words_path)
  15. default_textrank.set_stop_words(stop_words_path)