nltkor 1.2.16__tar.gz → 1.2.17__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (131) hide show
  1. {nltkor-1.2.16 → nltkor-1.2.17}/PKG-INFO +2 -8
  2. {nltkor-1.2.16 → nltkor-1.2.17}/README.md +3 -2
  3. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/__init__.py +1 -1
  4. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/search/faiss_search.py +9 -9
  5. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor.egg-info/PKG-INFO +2 -8
  6. {nltkor-1.2.16 → nltkor-1.2.17}/setup.py +1 -1
  7. {nltkor-1.2.16 → nltkor-1.2.17}/LICENSE.txt +0 -0
  8. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/Kor_char.py +0 -0
  9. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/alignment/__init__.py +0 -0
  10. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/cider/__init__.py +0 -0
  11. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/cider/cider.py +0 -0
  12. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/cider/cider_scorer.py +0 -0
  13. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/distance/__init__.py +0 -0
  14. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/distance/wasserstein.py +0 -0
  15. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/etc.py +0 -0
  16. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/lazyimport.py +0 -0
  17. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/make_requirement.py +0 -0
  18. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/__init__.py +0 -0
  19. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/bartscore.py +0 -0
  20. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/bertscore.py +0 -0
  21. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/bleu_tensor.py +0 -0
  22. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/classical.py +0 -0
  23. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/entment.py +0 -0
  24. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/eval.py +0 -0
  25. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/mauve.py +0 -0
  26. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/metrics/mauve_utils.py +0 -0
  27. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/misc/__init__.py +0 -0
  28. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/misc/string2string_basic_functions.py +0 -0
  29. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/misc/string2string_default_tokenizer.py +0 -0
  30. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/misc/string2string_hash_functions.py +0 -0
  31. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/misc/string2string_word_embeddings.py +0 -0
  32. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/search/__init__.py +0 -0
  33. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/search/classical.py +0 -0
  34. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/search/kobert_tokenizer.py +0 -0
  35. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__init__.py +0 -0
  36. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/__init__.cpython-38.pyc +0 -0
  37. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/__init__.cpython-39.pyc +0 -0
  38. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/sejong_download.cpython-38.pyc +0 -0
  39. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/sejong_download.cpython-39.pyc +0 -0
  40. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/ssem.cpython-38.pyc +0 -0
  41. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/__pycache__/ssem.cpython-39.pyc +0 -0
  42. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/ch.py +0 -0
  43. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/dict_semClassNum.txt +0 -0
  44. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/layer.txt +0 -0
  45. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/sejong_download.py +0 -0
  46. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/sejong/ssem.py +0 -0
  47. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/similarity/__init__.py +0 -0
  48. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/similarity/bartscore____.py +0 -0
  49. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/similarity/bertscore____.py +0 -0
  50. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/similarity/classical.py +0 -0
  51. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/similarity/cosine_similarity.py +0 -0
  52. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/__init__.py +0 -0
  53. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/__pycache__/__init__.cpython-38.pyc +0 -0
  54. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/__pycache__/__init__.cpython-39.pyc +0 -0
  55. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/__pycache__/espresso_tag.cpython-38.pyc +0 -0
  56. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/__pycache__/espresso_tag.cpython-39.pyc +0 -0
  57. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/espresso_tag.py +0 -0
  58. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__init__.py +0 -0
  59. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/__init__.cpython-38.pyc +0 -0
  60. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/__init__.cpython-39.pyc +0 -0
  61. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/attributes.cpython-38.pyc +0 -0
  62. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/attributes.cpython-39.pyc +0 -0
  63. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/config.cpython-38.pyc +0 -0
  64. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/config.cpython-39.pyc +0 -0
  65. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/metadata.cpython-38.pyc +0 -0
  66. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/metadata.cpython-39.pyc +0 -0
  67. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/reader.cpython-38.pyc +0 -0
  68. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/reader.cpython-39.pyc +0 -0
  69. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/taggers.cpython-38.pyc +0 -0
  70. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/taggers.cpython-39.pyc +0 -0
  71. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/utils.cpython-38.pyc +0 -0
  72. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/utils.cpython-39.pyc +0 -0
  73. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/word_dictionary.cpython-38.pyc +0 -0
  74. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/__pycache__/word_dictionary.cpython-39.pyc +0 -0
  75. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/arguments.py +0 -0
  76. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/attributes.py +0 -0
  77. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/config.py +0 -0
  78. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/metadata.py +0 -0
  79. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/__init__.py +0 -0
  80. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/__pycache__/__init__.cpython-38.pyc +0 -0
  81. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/__pycache__/__init__.cpython-39.pyc +0 -0
  82. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/__pycache__/ner_reader.cpython-38.pyc +0 -0
  83. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/__pycache__/ner_reader.cpython-39.pyc +0 -0
  84. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/macmorphoreader.py +0 -0
  85. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/ner/ner_reader.py +0 -0
  86. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/network.c +0 -0
  87. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/network.pyx +0 -0
  88. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/networkconv.pyx +0 -0
  89. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/networkdependencyconv.pyx +0 -0
  90. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/__init__.py +0 -0
  91. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/__pycache__/__init__.cpython-38.pyc +0 -0
  92. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/__pycache__/__init__.cpython-39.pyc +0 -0
  93. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/__pycache__/parse_reader.cpython-38.pyc +0 -0
  94. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/__pycache__/parse_reader.cpython-39.pyc +0 -0
  95. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/parse/parse_reader.py +0 -0
  96. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/__init__.py +0 -0
  97. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/__pycache__/__init__.cpython-38.pyc +0 -0
  98. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/__pycache__/__init__.cpython-39.pyc +0 -0
  99. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/__pycache__/pos_reader.cpython-38.pyc +0 -0
  100. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/__pycache__/pos_reader.cpython-39.pyc +0 -0
  101. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/macmorphoreader.py +0 -0
  102. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/pos/pos_reader.py +0 -0
  103. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/reader.py +0 -0
  104. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__init__.py +0 -0
  105. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/__init__.cpython-38.pyc +0 -0
  106. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/__init__.cpython-39.pyc +0 -0
  107. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/srl_reader.cpython-38.pyc +0 -0
  108. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/srl_reader.cpython-39.pyc +0 -0
  109. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/train_srl.cpython-38.pyc +0 -0
  110. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__pycache__/train_srl.cpython-39.pyc +0 -0
  111. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/__srl_reader_.py +0 -0
  112. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/srl_reader.py +0 -0
  113. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/srl/train_srl.py +0 -0
  114. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/taggers.py +0 -0
  115. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/utils.py +0 -0
  116. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/word_dictionary.py +0 -0
  117. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/__init__.py +0 -0
  118. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/__pycache__/__init__.cpython-38.pyc +0 -0
  119. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/__pycache__/__init__.cpython-39.pyc +0 -0
  120. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/__pycache__/wsd_reader.cpython-38.pyc +0 -0
  121. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/__pycache__/wsd_reader.cpython-39.pyc +0 -0
  122. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/macmorphoreader.py +0 -0
  123. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tag/libs/wsd/wsd_reader.py +0 -0
  124. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tokenize/__init__.py +0 -0
  125. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/tokenize/ko_tokenize.py +0 -0
  126. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor/trans.py +0 -0
  127. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor.egg-info/SOURCES.txt +0 -0
  128. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor.egg-info/dependency_links.txt +0 -0
  129. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor.egg-info/requires.txt +0 -0
  130. {nltkor-1.2.16 → nltkor-1.2.17}/nltkor.egg-info/top_level.txt +0 -0
  131. {nltkor-1.2.16 → nltkor-1.2.17}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.4
1
+ Metadata-Version: 2.1
2
2
  Name: nltkor
3
- Version: 1.2.16
3
+ Version: 1.2.17
4
4
  Home-page: https://modi.changwon.ac.kr/air_cwnu/nlp_tool/nltk_ko.git
5
5
  Keywords: string matching,pattern matching,edit distance,string to string correction,string to string matching,Levenshtein edit distance,Hamming distance,Damerau-Levenshtein distance,Jaro-Winkler distance,longest common subsequence,longest common substring,dynamic programming,approximate string matching,semantic similarity,natural language processing,NLP,information retrieval,rouge,sacrebleu,bertscore,bartscore,fasttext,glove,cosine similarity,Smith-Waterman,Needleman-Wunsch,Hirschberg,Karp-Rabin,Knuth-Morris-Pratt,Boyer-Moore
6
6
  Classifier: Programming Language :: Python :: 3.7
@@ -34,9 +34,3 @@ Requires-Dist: bert_score
34
34
  Requires-Dist: chardet
35
35
  Requires-Dist: GPUtil
36
36
  Requires-Dist: fasttext
37
- Dynamic: classifier
38
- Dynamic: home-page
39
- Dynamic: keywords
40
- Dynamic: license-file
41
- Dynamic: requires-dist
42
- Dynamic: requires-python
@@ -51,7 +51,7 @@
51
51
  - [5.11 ROUGE](#511-rouge)
52
52
  - [5.12 CIDER](#512-cider)
53
53
  - [5.13 METEOR](#513-meteor)
54
- - [5.14 EntMent](#514-entment)
54
+ - [5.14 EMR(Entity Mention Recall)](#514-emrentity-mention-recall)
55
55
  - [6 확장 평가 함수](#6-확장-평가-함수)
56
56
  - [6.1 MAUVE](#61-mauve)
57
57
  - [6.2 BERT Score](#62-bert-score)
@@ -915,10 +915,11 @@ TF-IDF를 n-gram에 대한 가중치로 계산하고 참조 캡션과 생성 캡
915
915
 
916
916
  요약된 텍스트가 참조 문서에 등장하는 중요 개체를 얼마나 잘 유지하고 있는지에 대한 평가 지표이다.
917
917
 
918
+ EMR().entity(원본 텍스트,요약된 텍스트)
918
919
  ```python
919
920
  >>> # -*- coding: utf-8 -*-
920
921
  >>> from nltkor.metrics import EntMent
921
- >>> EntMent().entity("국립창원대학교(총장 박민원)가 사천우주항공캠퍼스 개교와 함께 2025학년도 사천우주항공공학부 입학식을 7일 오전 11시 사천우주항공캠퍼스에서 열었다.이날 행사에는 박민원 총장을 비롯해 국민의힘 서천호 국회의원(사천·남해·하동), 윤영빈 우주항공청장, 박동식 사천시장, 김규헌 사천시의회 의장, 지역 유관기관 관계자들과 신입생 및 가족들이 참석했다. 글로컬대학30사업 선정에 따라 국립창원대와 통합을 추진 중인 경남도립거창대학, 경남도립남해대학 관계자도 함께 자리했다.행사는 1부 현판 제막식과 2부 입학식으로 진행됐으며, 박동식 사천시장은 신입생들에게 축하 선물로 금배지를 전달했고, 박민원 총장은 캠퍼스 설립에 기여한 유공자들에게 표창장을 수여했다.","국립창원대학교는 4월 7일 사천우주항공캠퍼스에서 2025학년도 사천우주항공공학부 입학식을 개최했다. 이날 행사에는 박민원 총장, 서천호 국회의원, 윤영빈 우주항공청장, 박동식 사천시장 등 주요 인사와 신입생 및 가족들이 참석했으며, 글로컬대학30사업과 관련된 거창대학·남해대학 관계자들도 함께했다. 행사는 현판 제막식과 입학식으로 나뉘어 진행되었고, 신입생들에게는 금배지가, 캠퍼스 설립 유공자들에게는 표창장이 수여되었다.")
922
+ >>> EMR().entity("국립창원대학교(총장 박민원)가 사천우주항공캠퍼스 개교와 함께 2025학년도 사천우주항공공학부 입학식을 7일 오전 11시 사천우주항공캠퍼스에서 열었다.이날 행사에는 박민원 총장을 비롯해 국민의힘 서천호 국회의원(사천·남해·하동), 윤영빈 우주항공청장, 박동식 사천시장, 김규헌 사천시의회 의장, 지역 유관기관 관계자들과 신입생 및 가족들이 참석했다. 글로컬대학30사업 선정에 따라 국립창원대와 통합을 추진 중인 경남도립거창대학, 경남도립남해대학 관계자도 함께 자리했다.행사는 1부 현판 제막식과 2부 입학식으로 진행됐으며, 박동식 사천시장은 신입생들에게 축하 선물로 금배지를 전달했고, 박민원 총장은 캠퍼스 설립에 기여한 유공자들에게 표창장을 수여했다.","국립창원대학교는 4월 7일 사천우주항공캠퍼스에서 2025학년도 사천우주항공공학부 입학식을 개최했다. 이날 행사에는 박민원 총장, 서천호 국회의원, 윤영빈 우주항공청장, 박동식 사천시장 등 주요 인사와 신입생 및 가족들이 참석했으며, 글로컬대학30사업과 관련된 거창대학·남해대학 관계자들도 함께했다. 행사는 현판 제막식과 입학식으로 나뉘어 진행되었고, 신입생들에게는 금배지가, 캠퍼스 설립 유공자들에게는 표창장이 수여되었다.")
922
923
  Downloading Espresso5 model...
923
924
  0.8888888888888888
924
925
  ```
@@ -13,4 +13,4 @@ from nltkor import trans
13
13
  from nltkor import Kor_char
14
14
  from nltkor import etc
15
15
 
16
- __version__ = '1.2.16'
16
+ __version__ = '1.2.17'
@@ -78,10 +78,10 @@ class FaissSearch:
78
78
  return FaissSearch_SenEmbed(model_name_or_path=model_name_or_path, embedding_type=embedding_type)
79
79
  elif mode == 'word':
80
80
  return FaissSearch_WordEmbed(model_name_or_path=model_name_or_path, embedding_type=embedding_type)
81
- elif mode == 'splade':
82
- return FaissSearch_Splade(model_name_or_path=model_name_or_path, embedding_type=embedding_type)
81
+ elif mode == 'sparse':
82
+ return FaissSearch_Sparse(model_name_or_path=model_name_or_path, embedding_type=embedding_type)
83
83
  else:
84
- raise ValueError("choice 'sentence' or 'word' or 'splade'")
84
+ raise ValueError("choice 'sentence' or 'word' or 'sparse'")
85
85
 
86
86
 
87
87
 
@@ -233,7 +233,7 @@ class FaissSearch_SenEmbed:
233
233
 
234
234
  # Get the embeddings
235
235
  with torch.no_grad():
236
- embeddings = self.model(**encoded_text)
236
+ embeddings = self.model(encoded_text['input_ids'])
237
237
 
238
238
  # Get the proper embedding type
239
239
  if embedding_type == 'last_hidden_state':
@@ -426,7 +426,7 @@ class FaissSearch_SenEmbed:
426
426
 
427
427
  # Return the dataset
428
428
  return self.dataset
429
-
429
+
430
430
 
431
431
  # Search for the most similar elements in the dataset, given a query
432
432
  def search(self,
@@ -465,6 +465,7 @@ class FaissSearch_SenEmbed:
465
465
  # Add the scores
466
466
  results_df['score'] = scores
467
467
 
468
+
468
469
  # Sort the results by score
469
470
  results_df.sort_values("score", ascending=True, inplace=True)
470
471
 
@@ -473,8 +474,7 @@ class FaissSearch_SenEmbed:
473
474
 
474
475
 
475
476
 
476
- # FAISS Splade + ICT library wrapper class
477
- class FaissSearch_Splade(FaissSearch_SenEmbed):
477
+ class FaissSearch_Sparse(FaissSearch_SenEmbed):
478
478
  def __init__(self,
479
479
  model_name_or_path: str = 'klue/bert-base',
480
480
  tokenizer_name_or_path: str = 'klue/bert-base',
@@ -580,14 +580,14 @@ class FaissSearch_Splade(FaissSearch_SenEmbed):
580
580
 
581
581
  # Get the embeddings
582
582
  with torch.no_grad():
583
- embeddings = self.model(**encoded_text)
583
+ embeddings = self.model(encoded_text['input_ids'])
584
584
 
585
585
  # Get the last hidden state
586
586
  embeddings = embeddings['logits']
587
587
 
588
588
  embeddings = torch.sum(torch.log(1+torch.relu(embeddings)) * encoded_text['attention_mask'].unsqueeze(-1), dim=1)
589
589
  e_norm = torch.nn.functional.normalize(embeddings, p=2, dim=1, eps=1e-8)
590
-
590
+
591
591
  # Return the embeddings
592
592
  return e_norm
593
593
 
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.4
1
+ Metadata-Version: 2.1
2
2
  Name: nltkor
3
- Version: 1.2.16
3
+ Version: 1.2.17
4
4
  Home-page: https://modi.changwon.ac.kr/air_cwnu/nlp_tool/nltk_ko.git
5
5
  Keywords: string matching,pattern matching,edit distance,string to string correction,string to string matching,Levenshtein edit distance,Hamming distance,Damerau-Levenshtein distance,Jaro-Winkler distance,longest common subsequence,longest common substring,dynamic programming,approximate string matching,semantic similarity,natural language processing,NLP,information retrieval,rouge,sacrebleu,bertscore,bartscore,fasttext,glove,cosine similarity,Smith-Waterman,Needleman-Wunsch,Hirschberg,Karp-Rabin,Knuth-Morris-Pratt,Boyer-Moore
6
6
  Classifier: Programming Language :: Python :: 3.7
@@ -34,9 +34,3 @@ Requires-Dist: bert_score
34
34
  Requires-Dist: chardet
35
35
  Requires-Dist: GPUtil
36
36
  Requires-Dist: fasttext
37
- Dynamic: classifier
38
- Dynamic: home-page
39
- Dynamic: keywords
40
- Dynamic: license-file
41
- Dynamic: requires-dist
42
- Dynamic: requires-python
@@ -66,7 +66,7 @@ module1 = cythonize([
66
66
 
67
67
  setup(
68
68
  name='nltkor',
69
- version='1.2.16',
69
+ version='1.2.17',
70
70
  url='https://modi.changwon.ac.kr/air_cwnu/nlp_tool/nltk_ko.git',
71
71
  packages=find_packages(exclude=[]),
72
72
  python_requires='>=3.7',
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes