nltk 3.2__zip → 3.3__zip

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 (406) hide show
  1. {nltk-3.2 → nltk-3.3}/INSTALL.txt +6 -6
  2. {nltk-3.2 → nltk-3.3}/LICENSE.txt +13 -13
  3. {nltk-3.2 → nltk-3.3}/MANIFEST.in +6 -6
  4. {nltk-3.2/nltk.egg-info → nltk-3.3}/PKG-INFO +13 -5
  5. nltk-3.2/README.txt → nltk-3.3/README.md +49 -48
  6. nltk-3.3/nltk/VERSION +1 -0
  7. {nltk-3.2 → nltk-3.3}/nltk/__init__.py +185 -185
  8. {nltk-3.2 → nltk-3.3}/nltk/app/__init__.py +52 -53
  9. {nltk-3.2 → nltk-3.3}/nltk/app/chartparser_app.py +2276 -2276
  10. {nltk-3.2 → nltk-3.3}/nltk/app/chunkparser_app.py +1262 -1264
  11. {nltk-3.2 → nltk-3.3}/nltk/app/collocations_app.py +347 -350
  12. {nltk-3.2 → nltk-3.3}/nltk/app/concordance_app.py +568 -570
  13. {nltk-3.2 → nltk-3.3}/nltk/app/nemo_app.py +156 -155
  14. {nltk-3.2 → nltk-3.3}/nltk/app/rdparser_app.py +892 -894
  15. {nltk-3.2 → nltk-3.3}/nltk/app/srparser_app.py +809 -809
  16. {nltk-3.2 → nltk-3.3}/nltk/app/wordfreq_app.py +32 -32
  17. {nltk-3.2 → nltk-3.3}/nltk/app/wordnet_app.py +968 -968
  18. {nltk-3.2 → nltk-3.3}/nltk/book.py +91 -90
  19. {nltk-3.2 → nltk-3.3}/nltk/ccg/__init__.py +22 -22
  20. {nltk-3.2 → nltk-3.3}/nltk/ccg/api.py +359 -335
  21. {nltk-3.2 → nltk-3.3}/nltk/ccg/chart.py +394 -361
  22. {nltk-3.2 → nltk-3.3}/nltk/ccg/combinator.py +343 -312
  23. {nltk-3.2 → nltk-3.3}/nltk/ccg/lexicon.py +328 -290
  24. nltk-3.3/nltk/ccg/logic.py +46 -0
  25. {nltk-3.2 → nltk-3.3}/nltk/chat/__init__.py +49 -49
  26. {nltk-3.2 → nltk-3.3}/nltk/chat/eliza.py +244 -244
  27. {nltk-3.2 → nltk-3.3}/nltk/chat/iesha.py +140 -140
  28. {nltk-3.2 → nltk-3.3}/nltk/chat/rude.py +92 -92
  29. {nltk-3.2 → nltk-3.3}/nltk/chat/suntsu.py +117 -117
  30. {nltk-3.2 → nltk-3.3}/nltk/chat/util.py +122 -120
  31. {nltk-3.2 → nltk-3.3}/nltk/chat/zen.py +282 -282
  32. {nltk-3.2 → nltk-3.3}/nltk/chunk/__init__.py +190 -190
  33. {nltk-3.2 → nltk-3.3}/nltk/chunk/api.py +51 -51
  34. {nltk-3.2 → nltk-3.3}/nltk/chunk/named_entity.py +331 -331
  35. {nltk-3.2 → nltk-3.3}/nltk/chunk/regexp.py +1390 -1384
  36. {nltk-3.2 → nltk-3.3}/nltk/chunk/util.py +600 -600
  37. {nltk-3.2 → nltk-3.3}/nltk/classify/__init__.py +98 -98
  38. {nltk-3.2 → nltk-3.3}/nltk/classify/api.py +193 -193
  39. {nltk-3.2 → nltk-3.3}/nltk/classify/decisiontree.py +295 -295
  40. {nltk-3.2 → nltk-3.3}/nltk/classify/maxent.py +1510 -1492
  41. {nltk-3.2 → nltk-3.3}/nltk/classify/megam.py +179 -178
  42. {nltk-3.2 → nltk-3.3}/nltk/classify/naivebayes.py +245 -242
  43. {nltk-3.2 → nltk-3.3}/nltk/classify/positivenaivebayes.py +170 -170
  44. {nltk-3.2 → nltk-3.3}/nltk/classify/rte_classify.py +160 -191
  45. {nltk-3.2 → nltk-3.3}/nltk/classify/scikitlearn.py +153 -151
  46. {nltk-3.2 → nltk-3.3}/nltk/classify/senna.py +183 -181
  47. {nltk-3.2 → nltk-3.3}/nltk/classify/svm.py +15 -15
  48. {nltk-3.2 → nltk-3.3}/nltk/classify/tadm.py +113 -112
  49. {nltk-3.2 → nltk-3.3}/nltk/classify/textcat.py +193 -193
  50. {nltk-3.2 → nltk-3.3}/nltk/classify/util.py +324 -312
  51. {nltk-3.2 → nltk-3.3}/nltk/classify/weka.py +346 -345
  52. {nltk-3.2 → nltk-3.3}/nltk/cluster/__init__.py +86 -86
  53. {nltk-3.2 → nltk-3.3}/nltk/cluster/api.py +74 -70
  54. {nltk-3.2 → nltk-3.3}/nltk/cluster/em.py +247 -247
  55. {nltk-3.2 → nltk-3.3}/nltk/cluster/gaac.py +168 -168
  56. {nltk-3.2 → nltk-3.3}/nltk/cluster/kmeans.py +221 -221
  57. {nltk-3.2 → nltk-3.3}/nltk/cluster/util.py +304 -289
  58. nltk-3.3/nltk/collections.py +640 -0
  59. {nltk-3.2 → nltk-3.3}/nltk/collocations.py +388 -388
  60. nltk-3.3/nltk/compat.py +365 -0
  61. {nltk-3.2 → nltk-3.3}/nltk/corpus/__init__.py +341 -320
  62. {nltk-3.2 → nltk-3.3}/nltk/corpus/europarl_raw.py +44 -44
  63. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/__init__.py +147 -145
  64. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/aligned.py +115 -114
  65. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/api.py +448 -446
  66. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bnc.py +252 -252
  67. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bracket_parse.py +227 -227
  68. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/categorized_sents.py +179 -178
  69. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chasen.py +140 -139
  70. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/childes.py +522 -494
  71. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chunked.py +212 -212
  72. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/cmudict.py +97 -95
  73. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/comparative_sents.py +280 -278
  74. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/conll.py +528 -521
  75. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/crubadan.py +116 -116
  76. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/dependency.py +101 -101
  77. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/framenet.py +3059 -2093
  78. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ieer.py +112 -111
  79. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/indian.py +85 -86
  80. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ipipan.py +332 -331
  81. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/knbc.py +187 -187
  82. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/lin.py +156 -156
  83. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/mte.py +311 -310
  84. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nkjp.py +429 -428
  85. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nombank.py +421 -421
  86. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nps_chat.py +73 -73
  87. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/opinion_lexicon.py +115 -115
  88. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/panlex_lite.py +165 -165
  89. nltk-3.3/nltk/corpus/reader/pl196x.py +292 -0
  90. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/plaintext.py +232 -233
  91. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ppattach.py +96 -95
  92. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/propbank.py +479 -481
  93. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/pros_cons.py +128 -126
  94. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/reviews.py +330 -328
  95. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/rte.py +144 -146
  96. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/semcor.py +256 -256
  97. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/senseval.py +203 -201
  98. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sentiwordnet.py +137 -137
  99. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sinica_treebank.py +75 -75
  100. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/string_category.py +60 -59
  101. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/switchboard.py +119 -119
  102. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/tagged.py +295 -294
  103. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/timit.py +452 -450
  104. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/toolbox.py +68 -68
  105. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/twitter.py +155 -157
  106. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/udhr.py +80 -80
  107. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/util.py +804 -800
  108. nltk-3.3/nltk/corpus/reader/verbnet.py +595 -0
  109. nltk-3.3/nltk/corpus/reader/wordlist.py +134 -0
  110. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/wordnet.py +2108 -2036
  111. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/xmldocs.py +392 -387
  112. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ycoe.py +243 -242
  113. {nltk-3.2 → nltk-3.3}/nltk/corpus/util.py +144 -127
  114. {nltk-3.2 → nltk-3.3}/nltk/data.py +1487 -1451
  115. {nltk-3.2 → nltk-3.3}/nltk/decorators.py +219 -215
  116. {nltk-3.2 → nltk-3.3}/nltk/downloader.py +2278 -2274
  117. {nltk-3.2 → nltk-3.3}/nltk/draw/__init__.py +27 -27
  118. {nltk-3.2 → nltk-3.3}/nltk/draw/cfg.py +774 -777
  119. {nltk-3.2 → nltk-3.3}/nltk/draw/dispersion.py +58 -58
  120. {nltk-3.2 → nltk-3.3}/nltk/draw/table.py +1098 -1098
  121. {nltk-3.2 → nltk-3.3}/nltk/draw/tree.py +963 -966
  122. {nltk-3.2 → nltk-3.3}/nltk/draw/util.py +2356 -2356
  123. {nltk-3.2 → nltk-3.3}/nltk/featstruct.py +2617 -2500
  124. {nltk-3.2 → nltk-3.3}/nltk/grammar.py +1559 -1516
  125. {nltk-3.2 → nltk-3.3}/nltk/help.py +56 -56
  126. {nltk-3.2 → nltk-3.3}/nltk/inference/__init__.py +20 -20
  127. {nltk-3.2 → nltk-3.3}/nltk/inference/api.py +597 -589
  128. {nltk-3.2 → nltk-3.3}/nltk/inference/discourse.py +612 -608
  129. {nltk-3.2 → nltk-3.3}/nltk/inference/mace.py +311 -311
  130. {nltk-3.2 → nltk-3.3}/nltk/inference/nonmonotonic.py +509 -509
  131. {nltk-3.2 → nltk-3.3}/nltk/inference/prover9.py +431 -431
  132. {nltk-3.2 → nltk-3.3}/nltk/inference/resolution.py +687 -687
  133. {nltk-3.2 → nltk-3.3}/nltk/inference/tableau.py +607 -607
  134. {nltk-3.2 → nltk-3.3}/nltk/internals.py +982 -982
  135. {nltk-3.2 → nltk-3.3}/nltk/jsontags.py +63 -63
  136. {nltk-3.2 → nltk-3.3}/nltk/lazyimport.py +142 -142
  137. {nltk-3.2 → nltk-3.3}/nltk/metrics/__init__.py +30 -29
  138. {nltk-3.2 → nltk-3.3}/nltk/metrics/agreement.py +421 -401
  139. nltk-3.3/nltk/metrics/aline.py +607 -0
  140. {nltk-3.2 → nltk-3.3}/nltk/metrics/association.py +414 -410
  141. {nltk-3.2 → nltk-3.3}/nltk/metrics/confusionmatrix.py +206 -206
  142. {nltk-3.2 → nltk-3.3}/nltk/metrics/distance.py +207 -202
  143. {nltk-3.2 → nltk-3.3}/nltk/metrics/paice.py +381 -381
  144. {nltk-3.2 → nltk-3.3}/nltk/metrics/scores.py +228 -228
  145. {nltk-3.2 → nltk-3.3}/nltk/metrics/segmentation.py +231 -232
  146. {nltk-3.2 → nltk-3.3}/nltk/metrics/spearman.py +69 -69
  147. {nltk-3.2 → nltk-3.3}/nltk/misc/__init__.py +11 -11
  148. {nltk-3.2 → nltk-3.3}/nltk/misc/babelfish.py +10 -10
  149. {nltk-3.2 → nltk-3.3}/nltk/misc/chomsky.py +135 -133
  150. {nltk-3.2 → nltk-3.3}/nltk/misc/minimalset.py +83 -83
  151. {nltk-3.2 → nltk-3.3}/nltk/misc/sort.py +157 -157
  152. {nltk-3.2 → nltk-3.3}/nltk/misc/wordfinder.py +129 -129
  153. {nltk-3.2 → nltk-3.3}/nltk/parse/__init__.py +81 -80
  154. {nltk-3.2 → nltk-3.3}/nltk/parse/api.py +66 -66
  155. {nltk-3.2 → nltk-3.3}/nltk/parse/bllip.py +282 -282
  156. {nltk-3.2 → nltk-3.3}/nltk/parse/chart.py +1682 -1681
  157. nltk-3.3/nltk/parse/corenlp.py +780 -0
  158. {nltk-3.2 → nltk-3.3}/nltk/parse/dependencygraph.py +760 -758
  159. {nltk-3.2 → nltk-3.3}/nltk/parse/earleychart.py +452 -451
  160. {nltk-3.2 → nltk-3.3}/nltk/parse/evaluate.py +129 -129
  161. {nltk-3.2 → nltk-3.3}/nltk/parse/featurechart.py +580 -578
  162. {nltk-3.2 → nltk-3.3}/nltk/parse/generate.py +87 -76
  163. {nltk-3.2 → nltk-3.3}/nltk/parse/malt.py +356 -356
  164. {nltk-3.2 → nltk-3.3}/nltk/parse/nonprojectivedependencyparser.py +783 -783
  165. {nltk-3.2 → nltk-3.3}/nltk/parse/pchart.py +522 -522
  166. {nltk-3.2 → nltk-3.3}/nltk/parse/projectivedependencyparser.py +579 -578
  167. {nltk-3.2 → nltk-3.3}/nltk/parse/recursivedescent.py +655 -656
  168. {nltk-3.2 → nltk-3.3}/nltk/parse/shiftreduce.py +458 -459
  169. {nltk-3.2 → nltk-3.3}/nltk/parse/stanford.py +431 -406
  170. {nltk-3.2 → nltk-3.3}/nltk/parse/transitionparser.py +774 -772
  171. {nltk-3.2 → nltk-3.3}/nltk/parse/util.py +231 -231
  172. {nltk-3.2 → nltk-3.3}/nltk/parse/viterbi.py +401 -401
  173. {nltk-3.2 → nltk-3.3}/nltk/probability.py +2410 -2355
  174. {nltk-3.2 → nltk-3.3}/nltk/sem/__init__.py +61 -61
  175. {nltk-3.2 → nltk-3.3}/nltk/sem/boxer.py +1261 -1261
  176. {nltk-3.2 → nltk-3.3}/nltk/sem/chat80.py +782 -782
  177. {nltk-3.2 → nltk-3.3}/nltk/sem/cooper_storage.py +118 -118
  178. {nltk-3.2 → nltk-3.3}/nltk/sem/drt.py +1258 -1257
  179. {nltk-3.2 → nltk-3.3}/nltk/sem/drt_glue_demo.py +485 -489
  180. {nltk-3.2 → nltk-3.3}/nltk/sem/evaluate.py +790 -784
  181. {nltk-3.2 → nltk-3.3}/nltk/sem/glue.py +696 -694
  182. {nltk-3.2 → nltk-3.3}/nltk/sem/hole.py +386 -384
  183. {nltk-3.2 → nltk-3.3}/nltk/sem/lfg.py +210 -208
  184. {nltk-3.2 → nltk-3.3}/nltk/sem/linearlogic.py +451 -449
  185. {nltk-3.2 → nltk-3.3}/nltk/sem/logic.py +1910 -1910
  186. {nltk-3.2 → nltk-3.3}/nltk/sem/relextract.py +476 -483
  187. {nltk-3.2 → nltk-3.3}/nltk/sem/skolemize.py +101 -101
  188. {nltk-3.2 → nltk-3.3}/nltk/sem/util.py +249 -249
  189. {nltk-3.2 → nltk-3.3}/nltk/sentiment/__init__.py +14 -14
  190. {nltk-3.2 → nltk-3.3}/nltk/sentiment/sentiment_analyzer.py +229 -229
  191. {nltk-3.2 → nltk-3.3}/nltk/sentiment/util.py +762 -752
  192. {nltk-3.2 → nltk-3.3}/nltk/sentiment/vader.py +443 -445
  193. {nltk-3.2 → nltk-3.3}/nltk/stem/__init__.py +31 -33
  194. {nltk-3.2 → nltk-3.3}/nltk/stem/api.py +28 -25
  195. nltk-3.3/nltk/stem/arlstem.py +369 -0
  196. {nltk-3.2 → nltk-3.3}/nltk/stem/isri.py +345 -345
  197. {nltk-3.2 → nltk-3.3}/nltk/stem/lancaster.py +337 -310
  198. nltk-3.3/nltk/stem/porter.py +710 -0
  199. {nltk-3.2 → nltk-3.3}/nltk/stem/regexp.py +61 -61
  200. {nltk-3.2 → nltk-3.3}/nltk/stem/rslp.py +142 -142
  201. {nltk-3.2 → nltk-3.3}/nltk/stem/snowball.py +4236 -3712
  202. {nltk-3.2 → nltk-3.3}/nltk/stem/util.py +22 -12
  203. {nltk-3.2 → nltk-3.3}/nltk/stem/wordnet.py +51 -51
  204. {nltk-3.2 → nltk-3.3}/nltk/tag/__init__.py +152 -127
  205. {nltk-3.2 → nltk-3.3}/nltk/tag/api.py +86 -81
  206. {nltk-3.2 → nltk-3.3}/nltk/tag/brill.py +424 -426
  207. {nltk-3.2 → nltk-3.3}/nltk/tag/brill_trainer.py +608 -608
  208. {nltk-3.2 → nltk-3.3}/nltk/tag/crf.py +204 -200
  209. {nltk-3.2 → nltk-3.3}/nltk/tag/hmm.py +1277 -1277
  210. {nltk-3.2 → nltk-3.3}/nltk/tag/hunpos.py +132 -132
  211. {nltk-3.2 → nltk-3.3}/nltk/tag/mapping.py +101 -101
  212. {nltk-3.2 → nltk-3.3}/nltk/tag/perceptron.py +329 -314
  213. {nltk-3.2 → nltk-3.3}/nltk/tag/senna.py +147 -145
  214. {nltk-3.2 → nltk-3.3}/nltk/tag/sequential.py +740 -736
  215. {nltk-3.2 → nltk-3.3}/nltk/tag/stanford.py +219 -196
  216. {nltk-3.2 → nltk-3.3}/nltk/tag/tnt.py +595 -604
  217. {nltk-3.2 → nltk-3.3}/nltk/tag/util.py +72 -72
  218. {nltk-3.2 → nltk-3.3}/nltk/tbl/__init__.py +28 -28
  219. {nltk-3.2 → nltk-3.3}/nltk/tbl/api.py +1 -1
  220. {nltk-3.2 → nltk-3.3}/nltk/tbl/demo.py +366 -366
  221. {nltk-3.2 → nltk-3.3}/nltk/tbl/erroranalysis.py +39 -39
  222. {nltk-3.2 → nltk-3.3}/nltk/tbl/feature.py +259 -261
  223. {nltk-3.2 → nltk-3.3}/nltk/tbl/rule.py +321 -321
  224. {nltk-3.2 → nltk-3.3}/nltk/tbl/template.py +303 -308
  225. {nltk-3.2 → nltk-3.3}/nltk/test/__init__.py +18 -18
  226. {nltk-3.2 → nltk-3.3}/nltk/test/all.py +24 -23
  227. {nltk-3.2 → nltk-3.3}/nltk/test/bleu.doctest +14 -14
  228. {nltk-3.2 → nltk-3.3}/nltk/test/bnc.doctest +60 -60
  229. {nltk-3.2 → nltk-3.3}/nltk/test/ccg.doctest +376 -395
  230. nltk-3.3/nltk/test/ccg_semantics.doctest +553 -0
  231. {nltk-3.2 → nltk-3.3}/nltk/test/chat80.doctest +234 -234
  232. {nltk-3.2 → nltk-3.3}/nltk/test/childes.doctest +184 -184
  233. {nltk-3.2 → nltk-3.3}/nltk/test/childes_fixt.py +14 -13
  234. {nltk-3.2 → nltk-3.3}/nltk/test/chunk.doctest +373 -373
  235. {nltk-3.2 → nltk-3.3}/nltk/test/classify.doctest +185 -185
  236. {nltk-3.2 → nltk-3.3}/nltk/test/classify_fixt.py +11 -10
  237. nltk-3.3/nltk/test/collections.doctest +20 -0
  238. {nltk-3.2 → nltk-3.3}/nltk/test/collocations.doctest +276 -276
  239. {nltk-3.2 → nltk-3.3}/nltk/test/compat.doctest +134 -138
  240. {nltk-3.2 → nltk-3.3}/nltk/test/compat_fixt.py +9 -8
  241. nltk-3.3/nltk/test/concordance.doctest +55 -0
  242. {nltk-3.2 → nltk-3.3}/nltk/test/corpus.doctest +2200 -2212
  243. {nltk-3.2 → nltk-3.3}/nltk/test/corpus_fixt.py +3 -3
  244. {nltk-3.2 → nltk-3.3}/nltk/test/crubadan.doctest +65 -65
  245. {nltk-3.2 → nltk-3.3}/nltk/test/data.doctest +379 -379
  246. {nltk-3.2 → nltk-3.3}/nltk/test/dependency.doctest +241 -241
  247. {nltk-3.2 → nltk-3.3}/nltk/test/discourse.doctest +546 -546
  248. {nltk-3.2 → nltk-3.3}/nltk/test/discourse_fixt.py +14 -13
  249. {nltk-3.2 → nltk-3.3}/nltk/test/doctest_nose_plugin.py +159 -156
  250. {nltk-3.2 → nltk-3.3}/nltk/test/drt.doctest +517 -517
  251. {nltk-3.2 → nltk-3.3}/nltk/test/featgram.doctest +607 -607
  252. {nltk-3.2 → nltk-3.3}/nltk/test/featstruct.doctest +1229 -1230
  253. {nltk-3.2 → nltk-3.3}/nltk/test/framenet.doctest +288 -239
  254. {nltk-3.2 → nltk-3.3}/nltk/test/generate.doctest +67 -67
  255. {nltk-3.2 → nltk-3.3}/nltk/test/gensim.doctest +140 -141
  256. {nltk-3.2 → nltk-3.3}/nltk/test/gensim_fixt.py +10 -9
  257. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics.doctest +384 -386
  258. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt.doctest +68 -68
  259. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt_fixt.py +12 -11
  260. {nltk-3.2 → nltk-3.3}/nltk/test/grammar.doctest +48 -48
  261. {nltk-3.2 → nltk-3.3}/nltk/test/grammartestsuites.doctest +109 -109
  262. {nltk-3.2 → nltk-3.3}/nltk/test/index.doctest +100 -100
  263. {nltk-3.2 → nltk-3.3}/nltk/test/inference.doctest +534 -534
  264. {nltk-3.2 → nltk-3.3}/nltk/test/inference_fixt.py +12 -11
  265. {nltk-3.2 → nltk-3.3}/nltk/test/internals.doctest +140 -140
  266. {nltk-3.2 → nltk-3.3}/nltk/test/japanese.doctest +48 -48
  267. {nltk-3.2 → nltk-3.3}/nltk/test/logic.doctest +1098 -1098
  268. {nltk-3.2 → nltk-3.3}/nltk/test/metrics.doctest +270 -270
  269. {nltk-3.2 → nltk-3.3}/nltk/test/misc.doctest +118 -118
  270. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic.doctest +286 -286
  271. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic_fixt.py +12 -11
  272. {nltk-3.2 → nltk-3.3}/nltk/test/paice.doctest +35 -35
  273. {nltk-3.2 → nltk-3.3}/nltk/test/parse.doctest +884 -884
  274. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en.doctest +565 -565
  275. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en_fixt.py +16 -16
  276. {nltk-3.2 → nltk-3.3}/nltk/test/probability.doctest +304 -304
  277. {nltk-3.2 → nltk-3.3}/nltk/test/probability_fixt.py +13 -12
  278. {nltk-3.2 → nltk-3.3}/nltk/test/propbank.doctest +176 -176
  279. {nltk-3.2 → nltk-3.3}/nltk/test/relextract.doctest +263 -263
  280. {nltk-3.2 → nltk-3.3}/nltk/test/resolution.doctest +221 -221
  281. {nltk-3.2 → nltk-3.3}/nltk/test/runtests.py +86 -64
  282. {nltk-3.2 → nltk-3.3}/nltk/test/segmentation_fixt.py +10 -10
  283. {nltk-3.2 → nltk-3.3}/nltk/test/semantics.doctest +665 -665
  284. {nltk-3.2 → nltk-3.3}/nltk/test/semantics_fixt.py +7 -7
  285. {nltk-3.2 → nltk-3.3}/nltk/test/sentiment.doctest +233 -233
  286. {nltk-3.2 → nltk-3.3}/nltk/test/sentiwordnet.doctest +41 -41
  287. {nltk-3.2 → nltk-3.3}/nltk/test/simple.doctest +85 -85
  288. {nltk-3.2 → nltk-3.3}/nltk/test/stem.doctest +78 -78
  289. {nltk-3.2 → nltk-3.3}/nltk/test/tag.doctest +33 -33
  290. {nltk-3.2 → nltk-3.3}/nltk/test/tokenize.doctest +273 -166
  291. {nltk-3.2 → nltk-3.3}/nltk/test/toolbox.doctest +307 -307
  292. {nltk-3.2 → nltk-3.3}/nltk/test/translate.doctest +242 -242
  293. {nltk-3.2 → nltk-3.3}/nltk/test/translate_fixt.py +3 -3
  294. {nltk-3.2 → nltk-3.3}/nltk/test/tree.doctest +1101 -1101
  295. {nltk-3.2 → nltk-3.3}/nltk/test/treeprettyprinter.doctest +127 -127
  296. {nltk-3.2 → nltk-3.3}/nltk/test/treetransforms.doctest +156 -156
  297. nltk-3.3/nltk/test/unit/test_2x_compat.py +63 -0
  298. nltk-3.3/nltk/test/unit/test_aline.py +52 -0
  299. nltk-3.3/nltk/test/unit/test_chunk.py +56 -0
  300. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_classify.py +55 -53
  301. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_collocations.py +88 -79
  302. nltk-3.3/nltk/test/unit/test_concordance.py +98 -0
  303. nltk-3.3/nltk/test/unit/test_corenlp.py +1000 -0
  304. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpora.py +200 -183
  305. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpus_views.py +45 -45
  306. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_hmm.py +88 -88
  307. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_json2csv_corpus.py +188 -187
  308. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_naivebayes.py +26 -26
  309. nltk-3.3/nltk/test/unit/test_rte_classify.py +86 -0
  310. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_seekable_unicode_stream_reader.py +124 -124
  311. nltk-3.3/nltk/test/unit/test_senna.py +74 -0
  312. nltk-3.3/nltk/test/unit/test_stem.py +121 -0
  313. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tag.py +20 -20
  314. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tgrep.py +628 -626
  315. nltk-3.3/nltk/test/unit/test_tokenize.py +168 -0
  316. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_twitter_auth.py +185 -185
  317. nltk-3.3/nltk/test/unit/test_wordnet.py +134 -0
  318. nltk-3.3/nltk/test/unit/translate/test_bleu.py +244 -0
  319. nltk-3.3/nltk/test/unit/translate/test_gdfa.py +57 -0
  320. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm1.py +75 -75
  321. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm2.py +87 -87
  322. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm3.py +105 -105
  323. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm4.py +126 -126
  324. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm5.py +166 -166
  325. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm_model.py +270 -270
  326. nltk-3.3/nltk/test/unit/translate/test_nist.py +37 -0
  327. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_stack_decoder.py +300 -300
  328. {nltk-3.2 → nltk-3.3}/nltk/test/unit/utils.py +41 -41
  329. {nltk-3.2 → nltk-3.3}/nltk/test/util.doctest +48 -48
  330. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet.doctest +595 -581
  331. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_fixt.py +7 -6
  332. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_lch.doctest +53 -53
  333. {nltk-3.2 → nltk-3.3}/nltk/test/wsd.doctest +68 -68
  334. {nltk-3.2 → nltk-3.3}/nltk/text.py +663 -610
  335. {nltk-3.2 → nltk-3.3}/nltk/tgrep.py +938 -937
  336. {nltk-3.2 → nltk-3.3}/nltk/tokenize/__init__.py +130 -107
  337. {nltk-3.2 → nltk-3.3}/nltk/tokenize/api.py +77 -78
  338. {nltk-3.2 → nltk-3.3}/nltk/tokenize/casual.py +343 -342
  339. {nltk-3.2 → nltk-3.3}/nltk/tokenize/mwe.py +120 -120
  340. nltk-3.3/nltk/tokenize/nist.py +173 -0
  341. {nltk-3.2 → nltk-3.3}/nltk/tokenize/punkt.py +1620 -1601
  342. {nltk-3.2 → nltk-3.3}/nltk/tokenize/regexp.py +208 -202
  343. nltk-3.3/nltk/tokenize/repp.py +151 -0
  344. {nltk-3.2 → nltk-3.3}/nltk/tokenize/sexpr.py +140 -142
  345. {nltk-3.2 → nltk-3.3}/nltk/tokenize/simple.py +138 -136
  346. {nltk-3.2 → nltk-3.3}/nltk/tokenize/stanford.py +117 -113
  347. nltk-3.3/nltk/tokenize/stanford_segmenter.py +246 -0
  348. {nltk-3.2 → nltk-3.3}/nltk/tokenize/texttiling.py +458 -460
  349. nltk-3.3/nltk/tokenize/toktok.py +155 -0
  350. nltk-3.3/nltk/tokenize/treebank.py +363 -0
  351. nltk-3.3/nltk/tokenize/util.py +261 -0
  352. {nltk-3.2 → nltk-3.3}/nltk/toolbox.py +499 -497
  353. {nltk-3.2 → nltk-3.3}/nltk/translate/__init__.py +24 -24
  354. {nltk-3.2 → nltk-3.3}/nltk/translate/api.py +331 -321
  355. {nltk-3.2 → nltk-3.3}/nltk/translate/bleu_score.py +615 -556
  356. nltk-3.3/nltk/translate/chrf_score.py +137 -0
  357. {nltk-3.2 → nltk-3.3}/nltk/translate/gale_church.py +248 -235
  358. {nltk-3.2 → nltk-3.3}/nltk/translate/gdfa.py +135 -131
  359. nltk-3.3/nltk/translate/gleu_score.py +193 -0
  360. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm1.py +250 -241
  361. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm2.py +308 -308
  362. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm3.py +337 -337
  363. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm4.py +467 -467
  364. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm5.py +639 -639
  365. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm_model.py +536 -533
  366. {nltk-3.2 → nltk-3.3}/nltk/translate/metrics.py +40 -40
  367. nltk-3.3/nltk/translate/nist_score.py +178 -0
  368. {nltk-3.2 → nltk-3.3}/nltk/translate/phrase_based.py +175 -175
  369. {nltk-3.2 → nltk-3.3}/nltk/translate/ribes_score.py +325 -325
  370. {nltk-3.2 → nltk-3.3}/nltk/translate/stack_decoder.py +499 -495
  371. {nltk-3.2 → nltk-3.3}/nltk/tree.py +1605 -1603
  372. {nltk-3.2 → nltk-3.3}/nltk/treeprettyprinter.py +564 -564
  373. {nltk-3.2 → nltk-3.3}/nltk/treetransforms.py +309 -309
  374. {nltk-3.2 → nltk-3.3}/nltk/twitter/__init__.py +28 -28
  375. {nltk-3.2 → nltk-3.3}/nltk/twitter/api.py +141 -136
  376. {nltk-3.2 → nltk-3.3}/nltk/twitter/common.py +264 -258
  377. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitter_demo.py +288 -288
  378. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitterclient.py +530 -530
  379. {nltk-3.2 → nltk-3.3}/nltk/twitter/util.py +147 -146
  380. nltk-3.3/nltk/util.py +751 -0
  381. {nltk-3.2 → nltk-3.3}/nltk/wsd.py +53 -53
  382. {nltk-3.2 → nltk-3.3/nltk.egg-info}/PKG-INFO +13 -5
  383. {nltk-3.2 → nltk-3.3}/nltk.egg-info/SOURCES.txt +26 -3
  384. nltk-3.3/nltk.egg-info/requires.txt +31 -0
  385. {nltk-3.2 → nltk-3.3}/setup.cfg +0 -1
  386. {nltk-3.2 → nltk-3.3}/setup.py +102 -75
  387. nltk-3.2/nltk/VERSION +0 -1
  388. nltk-3.2/nltk/compat.py +0 -680
  389. nltk-3.2/nltk/corpus/reader/pl196x.py +0 -278
  390. nltk-3.2/nltk/corpus/reader/verbnet.py +0 -408
  391. nltk-3.2/nltk/corpus/reader/wordlist.py +0 -37
  392. nltk-3.2/nltk/sentiment/vader_lexicon.txt +0 -7517
  393. nltk-3.2/nltk/six.py +0 -868
  394. nltk-3.2/nltk/stem/porter.py +0 -692
  395. nltk-3.2/nltk/test/unit/test_2x_compat.py +0 -26
  396. nltk-3.2/nltk/test/unit/test_stem.py +0 -36
  397. nltk-3.2/nltk/test/unit/test_tokenize.py +0 -23
  398. nltk-3.2/nltk/test/unit/translate/test_bleu.py +0 -84
  399. nltk-3.2/nltk/tokenize/treebank.py +0 -124
  400. nltk-3.2/nltk/tokenize/util.py +0 -90
  401. nltk-3.2/nltk/util.py +0 -1364
  402. {nltk-3.2 → nltk-3.3}/nltk/test/unit/__init__.py +0 -0
  403. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/__init__.py +0 -0
  404. {nltk-3.2 → nltk-3.3}/nltk.egg-info/dependency_links.txt +0 -0
  405. {nltk-3.2 → nltk-3.3}/nltk.egg-info/not-zip-safe +0 -0
  406. {nltk-3.2 → nltk-3.3}/nltk.egg-info/top_level.txt +0 -0
@@ -1,40 +1,40 @@
1
- # Natural Language Toolkit: Translation metrics
2
- #
3
- # Copyright (C) 2001-2015 NLTK Project
4
- # Author: Will Zhang <wilzzha@gmail.com>
5
- # Guan Gui <ggui@student.unimelb.edu.au>
6
- # Steven Bird <stevenbird1@gmail.com>
7
- # URL: <http://nltk.org/>
8
- # For license information, see LICENSE.TXT
9
- from __future__ import division
10
-
11
- def alignment_error_rate(reference, hypothesis, possible=None):
12
- """
13
- Return the Alignment Error Rate (AER) of an alignment
14
- with respect to a "gold standard" reference alignment.
15
- Return an error rate between 0.0 (perfect alignment) and 1.0 (no
16
- alignment).
17
-
18
- >>> from nltk.translate import Alignment
19
- >>> ref = Alignment([(0, 0), (1, 1), (2, 2)])
20
- >>> test = Alignment([(0, 0), (1, 2), (2, 1)])
21
- >>> alignment_error_rate(ref, test) # doctest: +ELLIPSIS
22
- 0.6666666666666667
23
-
24
- :type reference: Alignment
25
- :param reference: A gold standard alignment (sure alignments)
26
- :type hypothesis: Alignment
27
- :param hypothesis: A hypothesis alignment (aka. candidate alignments)
28
- :type possible: Alignment or None
29
- :param possible: A gold standard reference of possible alignments
30
- (defaults to *reference* if None)
31
- :rtype: float or None
32
- """
33
-
34
- if possible is None:
35
- possible = reference
36
- else:
37
- assert(reference.issubset(possible)) # sanity check
38
-
39
- return (1.0 - (len(hypothesis & reference) + len(hypothesis & possible)) /
40
- float(len(hypothesis) + len(reference)))
1
+ # Natural Language Toolkit: Translation metrics
2
+ #
3
+ # Copyright (C) 2001-2018 NLTK Project
4
+ # Author: Will Zhang <wilzzha@gmail.com>
5
+ # Guan Gui <ggui@student.unimelb.edu.au>
6
+ # Steven Bird <stevenbird1@gmail.com>
7
+ # URL: <http://nltk.org/>
8
+ # For license information, see LICENSE.TXT
9
+ from __future__ import division
10
+
11
+ def alignment_error_rate(reference, hypothesis, possible=None):
12
+ """
13
+ Return the Alignment Error Rate (AER) of an alignment
14
+ with respect to a "gold standard" reference alignment.
15
+ Return an error rate between 0.0 (perfect alignment) and 1.0 (no
16
+ alignment).
17
+
18
+ >>> from nltk.translate import Alignment
19
+ >>> ref = Alignment([(0, 0), (1, 1), (2, 2)])
20
+ >>> test = Alignment([(0, 0), (1, 2), (2, 1)])
21
+ >>> alignment_error_rate(ref, test) # doctest: +ELLIPSIS
22
+ 0.6666666666666667
23
+
24
+ :type reference: Alignment
25
+ :param reference: A gold standard alignment (sure alignments)
26
+ :type hypothesis: Alignment
27
+ :param hypothesis: A hypothesis alignment (aka. candidate alignments)
28
+ :type possible: Alignment or None
29
+ :param possible: A gold standard reference of possible alignments
30
+ (defaults to *reference* if None)
31
+ :rtype: float or None
32
+ """
33
+
34
+ if possible is None:
35
+ possible = reference
36
+ else:
37
+ assert(reference.issubset(possible)) # sanity check
38
+
39
+ return (1.0 - (len(hypothesis & reference) + len(hypothesis & possible)) /
40
+ float(len(hypothesis) + len(reference)))
@@ -0,0 +1,178 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Natural Language Toolkit: NIST Score
3
+ #
4
+ # Copyright (C) 2001-2018 NLTK Project
5
+ # Authors:
6
+ # Contributors:
7
+ # URL: <http://nltk.org/>
8
+ # For license information, see LICENSE.TXT
9
+
10
+ """NIST score implementation."""
11
+ from __future__ import division
12
+
13
+ import math
14
+ import fractions
15
+ from collections import Counter
16
+
17
+ from nltk.util import ngrams
18
+
19
+ def sentence_nist(references, hypothesis, n=5):
20
+ """
21
+ Calculate NIST score from
22
+ George Doddington. 2002. "Automatic evaluation of machine translation quality
23
+ using n-gram co-occurrence statistics." Proceedings of HLT.
24
+ Morgan Kaufmann Publishers Inc. http://dl.acm.org/citation.cfm?id=1289189.1289273
25
+
26
+ DARPA commissioned NIST to develop an MT evaluation facility based on the BLEU
27
+ score. The official script used by NIST to compute BLEU and NIST score is
28
+ mteval-14.pl. The main differences are:
29
+
30
+ - BLEU uses geometric mean of the ngram overlaps, NIST uses arithmetic mean.
31
+ - NIST has a different brevity penalty
32
+ - NIST score from mteval-14.pl has a self-contained tokenizer
33
+
34
+ Note: The mteval-14.pl includes a smoothing function for BLEU score that is NOT
35
+ used in the NIST score computation.
36
+
37
+ >>> hypothesis1 = ['It', 'is', 'a', 'guide', 'to', 'action', 'which',
38
+ ... 'ensures', 'that', 'the', 'military', 'always',
39
+ ... 'obeys', 'the', 'commands', 'of', 'the', 'party']
40
+
41
+ >>> hypothesis2 = ['It', 'is', 'to', 'insure', 'the', 'troops',
42
+ ... 'forever', 'hearing', 'the', 'activity', 'guidebook',
43
+ ... 'that', 'party', 'direct']
44
+
45
+ >>> reference1 = ['It', 'is', 'a', 'guide', 'to', 'action', 'that',
46
+ ... 'ensures', 'that', 'the', 'military', 'will', 'forever',
47
+ ... 'heed', 'Party', 'commands']
48
+
49
+ >>> reference2 = ['It', 'is', 'the', 'guiding', 'principle', 'which',
50
+ ... 'guarantees', 'the', 'military', 'forces', 'always',
51
+ ... 'being', 'under', 'the', 'command', 'of', 'the',
52
+ ... 'Party']
53
+
54
+ >>> reference3 = ['It', 'is', 'the', 'practical', 'guide', 'for', 'the',
55
+ ... 'army', 'always', 'to', 'heed', 'the', 'directions',
56
+ ... 'of', 'the', 'party']
57
+
58
+ >>> sentence_nist([reference1, reference2, reference3], hypothesis1) # doctest: +ELLIPSIS
59
+ 3.3709...
60
+
61
+ >>> sentence_nist([reference1, reference2, reference3], hypothesis2) # doctest: +ELLIPSIS
62
+ 1.4619...
63
+
64
+ :param references: reference sentences
65
+ :type references: list(list(str))
66
+ :param hypothesis: a hypothesis sentence
67
+ :type hypothesis: list(str)
68
+ :param n: highest n-gram order
69
+ :type n: int
70
+ """
71
+ return corpus_nist([references], [hypothesis], n)
72
+
73
+ def corpus_nist(list_of_references, hypotheses, n=5):
74
+ """
75
+ Calculate a single corpus-level NIST score (aka. system-level BLEU) for all
76
+ the hypotheses and their respective references.
77
+
78
+ :param references: a corpus of lists of reference sentences, w.r.t. hypotheses
79
+ :type references: list(list(list(str)))
80
+ :param hypotheses: a list of hypothesis sentences
81
+ :type hypotheses: list(list(str))
82
+ :param n: highest n-gram order
83
+ :type n: int
84
+ """
85
+ # Before proceeding to compute NIST, perform sanity checks.
86
+ assert len(list_of_references) == len(hypotheses), "The number of hypotheses and their reference(s) should be the same"
87
+
88
+ # Collect the ngram coounts from the reference sentences.
89
+ ngram_freq = Counter()
90
+ total_reference_words = 0
91
+ for references in list_of_references: # For each source sent, there's a list of reference sents.
92
+ for reference in references:
93
+ # For each order of ngram, count the ngram occurrences.
94
+ for i in range(1,n+1):
95
+ ngram_freq.update(ngrams(reference, i))
96
+ total_reference_words += len(reference)
97
+
98
+ # Compute the information weights based on the reference sentences.
99
+ # Eqn 2 in Doddington (2002):
100
+ # Info(w_1 ... w_n) = log_2 [ (# of occurrences of w_1 ... w_n-1) / (# of occurrences of w_1 ... w_n) ]
101
+ information_weights = {}
102
+ for _ngram in ngram_freq: # w_1 ... w_n
103
+ _mgram = _ngram[:-1] # w_1 ... w_n-1
104
+ # From https://github.com/moses-smt/mosesdecoder/blob/master/scripts/generic/mteval-v13a.pl#L546
105
+ # it's computed as such:
106
+ # denominator = ngram_freq[_mgram] if _mgram and _mgram in ngram_freq else denominator = total_reference_words
107
+ # information_weights[_ngram] = -1 * math.log(ngram_freq[_ngram]/denominator) / math.log(2)
108
+ #
109
+ # Mathematically, it's equivalent to the our implementation:
110
+ if _mgram and _mgram in ngram_freq:
111
+ numerator = ngram_freq[_mgram]
112
+ else:
113
+ numerator = total_reference_words
114
+ information_weights[_ngram] = math.log(numerator/ngram_freq[_ngram], 2)
115
+
116
+ # Micro-average.
117
+ nist_precision_numerator_per_ngram = Counter()
118
+ nist_precision_denominator_per_ngram = Counter()
119
+ l_ref, l_sys = 0, 0
120
+ # For each order of ngram.
121
+ for i in range(1,n+1):
122
+ # Iterate through each hypothesis and their corresponding references.
123
+ for references, hypothesis in zip(list_of_references, hypotheses):
124
+ hyp_len = len(hypothesis)
125
+
126
+ # Find reference with the best NIST score.
127
+ nist_score_per_ref = []
128
+ for reference in references:
129
+ _ref_len = len(reference)
130
+ # Counter of ngrams in hypothesis.
131
+ hyp_ngrams = Counter(ngrams(hypothesis, i)) if len(hypothesis) >= i else Counter()
132
+ ref_ngrams = Counter(ngrams(reference, i)) if len(reference) >=i else Counter()
133
+ ngram_overlaps = hyp_ngrams & ref_ngrams
134
+ # Precision part of the score in Eqn 3
135
+ _numerator = sum(information_weights[_ngram] * count
136
+ for _ngram, count in ngram_overlaps.items())
137
+ _denominator = sum(hyp_ngrams.values())
138
+ _precision = 0 if _denominator == 0 else _numerator/_denominator
139
+ nist_score_per_ref.append((_precision, _numerator, _denominator, _ref_len))
140
+ # Best reference.
141
+ precision, numerator, denominator, ref_len = max(nist_score_per_ref)
142
+ nist_precision_numerator_per_ngram[i] += numerator
143
+ nist_precision_denominator_per_ngram[i] += denominator
144
+ l_ref += ref_len
145
+ l_sys += hyp_len
146
+
147
+ # Final NIST micro-average mean aggregation.
148
+ nist_precision = 0
149
+ for i in nist_precision_numerator_per_ngram:
150
+ precision = nist_precision_numerator_per_ngram[i] / nist_precision_denominator_per_ngram[i]
151
+ nist_precision+= precision
152
+ # Eqn 3 in Doddington(2002)
153
+ return nist_precision * nist_length_penalty(l_ref, l_sys)
154
+
155
+
156
+ def nist_length_penalty(ref_len, hyp_len):
157
+ """
158
+ Calculates the NIST length penalty, from Eq. 3 in Doddington (2002)
159
+
160
+ penalty = exp( beta * log( min( len(hyp)/len(ref) , 1.0 )))
161
+
162
+ where,
163
+
164
+ `beta` is chosen to make the brevity penalty factor = 0.5 when the
165
+ no. of words in the system output (hyp) is 2/3 of the average
166
+ no. of words in the reference translation (ref)
167
+
168
+ The NIST penalty is different from BLEU's such that it minimize the impact
169
+ of the score of small variations in the length of a translation.
170
+ See Fig. 4 in Doddington (2002)
171
+ """
172
+ ratio = hyp_len / ref_len
173
+ if 0 < ratio < 1:
174
+ ratio_x, score_x = 1.5, 0.5
175
+ beta = math.log(score_x) / math.log(ratio_x)**2
176
+ return math.exp(beta * math.log(ratio)**2)
177
+ else: # ratio <= 0 or ratio >= 1
178
+ return max(min(ratio, 1.0), 0.0)
@@ -1,175 +1,175 @@
1
- # -*- coding: utf-8 -*-
2
- # Natural Language Toolkit: Phrase Extraction Algorithm
3
- #
4
- # Copyright (C) 2001-2015 NLTK Project
5
- # Authors: Liling Tan, Fredrik Hedman, Petra Barancikova
6
- # URL: <http://nltk.org/>
7
- # For license information, see LICENSE.TXT
8
-
9
- def extract(f_start, f_end, e_start, e_end,
10
- alignment, f_aligned,
11
- srctext, trgtext, srclen, trglen, max_phrase_length):
12
- """
13
- This function checks for alignment point consistency and extracts
14
- phrases using the chunk of consistent phrases.
15
-
16
- A phrase pair (e, f ) is consistent with an alignment A if and only if:
17
-
18
- (i) No English words in the phrase pair are aligned to words outside it.
19
-
20
- ∀e i ∈ e, (e i , f j ) ∈ A ⇒ f j ∈ f
21
-
22
- (ii) No Foreign words in the phrase pair are aligned to words outside it.
23
-
24
- ∀f j ∈ f , (e i , f j ) ∈ A ⇒ e i ∈ e
25
-
26
- (iii) The phrase pair contains at least one alignment point.
27
-
28
- ∃e i ∈ e ̄ , f j ∈ f ̄ s.t. (e i , f j ) ∈ A
29
-
30
- :type f_start: int
31
- :param f_start: Starting index of the possible foreign language phrases
32
- :type f_end: int
33
- :param f_end: Starting index of the possible foreign language phrases
34
- :type e_start: int
35
- :param e_start: Starting index of the possible source language phrases
36
- :type e_end: int
37
- :param e_end: Starting index of the possible source language phrases
38
- :type srctext: list
39
- :param srctext: The source language tokens, a list of string.
40
- :type trgtext: list
41
- :param trgtext: The target language tokens, a list of string.
42
- :type srclen: int
43
- :param srclen: The number of tokens in the source language tokens.
44
- :type trglen: int
45
- :param trglen: The number of tokens in the target language tokens.
46
- """
47
-
48
- if f_end < 0: # 0-based indexing.
49
- return {}
50
- # Check if alignment points are consistent.
51
- for e,f in alignment:
52
- if ((f_start <= f <= f_end) and (e < e_start or e > e_end)):
53
- return {}
54
-
55
- # Add phrase pairs (incl. additional unaligned f)
56
- phrases = set()
57
- fs = f_start
58
- while True:
59
- fe = min(f_end, f_start + max_phrase_length - 1)
60
- while True:
61
- # add phrase pair ([e_start, e_end], [fs, fe]) to set E
62
- # Need to +1 in range to include the end-point.
63
- src_phrase = " ".join(srctext[e_start:e_end+1])
64
- trg_phrase = " ".join(trgtext[fs:fe+1])
65
- # Include more data for later ordering.
66
- phrases.add(((e_start, e_end+1), (f_start, f_end+1),
67
- src_phrase, trg_phrase))
68
- fe += 1
69
- if fe in f_aligned or fe == trglen:
70
- break
71
- fs -=1
72
- if fs in f_aligned or fs < 0:
73
- break
74
- return phrases
75
-
76
- def phrase_extraction(srctext, trgtext, alignment, max_phrase_length=0):
77
- """
78
- Phrase extraction algorithm extracts all consistent phrase pairs from
79
- a word-aligned sentence pair.
80
-
81
- The idea is to loop over all possible source language (e) phrases and find
82
- the minimal foreign phrase (f) that matches each of them. Matching is done
83
- by identifying all alignment points for the source phrase and finding the
84
- shortest foreign phrase that includes all the foreign counterparts for the
85
- source words.
86
-
87
- In short, a phrase alignment has to
88
- (a) contain all alignment points for all covered words
89
- (b) contain at least one alignment point
90
-
91
- >>> srctext = "michael assumes that he will stay in the house"
92
- >>> trgtext = "michael geht davon aus , dass er im haus bleibt"
93
- >>> alignment = [(0,0), (1,1), (1,2), (1,3), (2,5), (3,6), (4,9),
94
- ... (5,9), (6,7), (7,7), (8,8)]
95
- >>> phrases = phrase_extraction(srctext, trgtext, alignment)
96
- >>> for i in sorted(phrases):
97
- ... print(i)
98
- ...
99
- ((0, 1), (0, 1), 'michael', 'michael')
100
- ((0, 2), (0, 4), 'michael assumes', 'michael geht davon aus')
101
- ((0, 2), (0, 4), 'michael assumes', 'michael geht davon aus ,')
102
- ((0, 3), (0, 6), 'michael assumes that', 'michael geht davon aus , dass')
103
- ((0, 4), (0, 7), 'michael assumes that he', 'michael geht davon aus , dass er')
104
- ((0, 9), (0, 10), 'michael assumes that he will stay in the house', 'michael geht davon aus , dass er im haus bleibt')
105
- ((1, 2), (1, 4), 'assumes', 'geht davon aus')
106
- ((1, 2), (1, 4), 'assumes', 'geht davon aus ,')
107
- ((1, 3), (1, 6), 'assumes that', 'geht davon aus , dass')
108
- ((1, 4), (1, 7), 'assumes that he', 'geht davon aus , dass er')
109
- ((1, 9), (1, 10), 'assumes that he will stay in the house', 'geht davon aus , dass er im haus bleibt')
110
- ((2, 3), (5, 6), 'that', ', dass')
111
- ((2, 3), (5, 6), 'that', 'dass')
112
- ((2, 4), (5, 7), 'that he', ', dass er')
113
- ((2, 4), (5, 7), 'that he', 'dass er')
114
- ((2, 9), (5, 10), 'that he will stay in the house', ', dass er im haus bleibt')
115
- ((2, 9), (5, 10), 'that he will stay in the house', 'dass er im haus bleibt')
116
- ((3, 4), (6, 7), 'he', 'er')
117
- ((3, 9), (6, 10), 'he will stay in the house', 'er im haus bleibt')
118
- ((4, 6), (9, 10), 'will stay', 'bleibt')
119
- ((4, 9), (7, 10), 'will stay in the house', 'im haus bleibt')
120
- ((6, 8), (7, 8), 'in the', 'im')
121
- ((6, 9), (7, 9), 'in the house', 'im haus')
122
- ((8, 9), (8, 9), 'house', 'haus')
123
-
124
- :type srctext: str
125
- :param srctext: The sentence string from the source language.
126
- :type trgtext: str
127
- :param trgtext: The sentence string from the target language.
128
- :type alignment: str
129
- :param alignment: The word alignment outputs as list of tuples, where
130
- the first elements of tuples are the source words' indices and
131
- second elements are the target words' indices. This is also the output
132
- format of nltk.translate.ibm1
133
- :rtype: list(tuple)
134
- :return: A list of tuples, each element in a list is a phrase and each
135
- phrase is a tuple made up of (i) its source location, (ii) its target
136
- location, (iii) the source phrase and (iii) the target phrase. The phrase
137
- list of tuples represents all the possible phrases extracted from the
138
- word alignments.
139
- :type max_phrase_length: int
140
- :param max_phrase_length: maximal phrase length, if 0 or not specified
141
- it is set to a length of the longer sentence (srctext or trgtext).
142
- """
143
-
144
- srctext = srctext.split() # e
145
- trgtext = trgtext.split() # f
146
- srclen = len(srctext) # len(e)
147
- trglen = len(trgtext) # len(f)
148
- # Keeps an index of which source/target words that are aligned.
149
- f_aligned = [j for _,j in alignment]
150
- max_phrase_length = max_phrase_length or max(srclen,trglen)
151
-
152
- # set of phrase pairs BP
153
- bp = set()
154
-
155
- for e_start in range(srclen):
156
- max_idx = min(srclen, e_start + max_phrase_length)
157
- for e_end in range(e_start, max_idx):
158
- # // find the minimally matching foreign phrase
159
- # (f start , f end ) = ( length(f), 0 )
160
- # f_start ∈ [0, len(f) - 1]; f_end ∈ [0, len(f) - 1]
161
- f_start, f_end = trglen-1 , -1 # 0-based indexing
162
-
163
- for e,f in alignment:
164
- if e_start <= e <= e_end:
165
- f_start = min(f, f_start)
166
- f_end = max(f, f_end)
167
- # add extract (f start , f end , e start , e end ) to set BP
168
- phrases = extract(f_start, f_end, e_start, e_end,
169
- alignment, f_aligned,
170
- srctext, trgtext, srclen, trglen,
171
- max_phrase_length)
172
- if phrases:
173
- bp.update(phrases)
174
- return bp
175
-
1
+ # -*- coding: utf-8 -*-
2
+ # Natural Language Toolkit: Phrase Extraction Algorithm
3
+ #
4
+ # Copyright (C) 2001-2018 NLTK Project
5
+ # Authors: Liling Tan, Fredrik Hedman, Petra Barancikova
6
+ # URL: <http://nltk.org/>
7
+ # For license information, see LICENSE.TXT
8
+
9
+ def extract(f_start, f_end, e_start, e_end,
10
+ alignment, f_aligned,
11
+ srctext, trgtext, srclen, trglen, max_phrase_length):
12
+ """
13
+ This function checks for alignment point consistency and extracts
14
+ phrases using the chunk of consistent phrases.
15
+
16
+ A phrase pair (e, f ) is consistent with an alignment A if and only if:
17
+
18
+ (i) No English words in the phrase pair are aligned to words outside it.
19
+
20
+ ∀e i ∈ e, (e i , f j ) ∈ A ⇒ f j ∈ f
21
+
22
+ (ii) No Foreign words in the phrase pair are aligned to words outside it.
23
+
24
+ ∀f j ∈ f , (e i , f j ) ∈ A ⇒ e i ∈ e
25
+
26
+ (iii) The phrase pair contains at least one alignment point.
27
+
28
+ ∃e i ∈ e ̄ , f j ∈ f ̄ s.t. (e i , f j ) ∈ A
29
+
30
+ :type f_start: int
31
+ :param f_start: Starting index of the possible foreign language phrases
32
+ :type f_end: int
33
+ :param f_end: Starting index of the possible foreign language phrases
34
+ :type e_start: int
35
+ :param e_start: Starting index of the possible source language phrases
36
+ :type e_end: int
37
+ :param e_end: Starting index of the possible source language phrases
38
+ :type srctext: list
39
+ :param srctext: The source language tokens, a list of string.
40
+ :type trgtext: list
41
+ :param trgtext: The target language tokens, a list of string.
42
+ :type srclen: int
43
+ :param srclen: The number of tokens in the source language tokens.
44
+ :type trglen: int
45
+ :param trglen: The number of tokens in the target language tokens.
46
+ """
47
+
48
+ if f_end < 0: # 0-based indexing.
49
+ return {}
50
+ # Check if alignment points are consistent.
51
+ for e,f in alignment:
52
+ if ((f_start <= f <= f_end) and (e < e_start or e > e_end)):
53
+ return {}
54
+
55
+ # Add phrase pairs (incl. additional unaligned f)
56
+ phrases = set()
57
+ fs = f_start
58
+ while True:
59
+ fe = min(f_end, f_start + max_phrase_length - 1)
60
+ while True:
61
+ # add phrase pair ([e_start, e_end], [fs, fe]) to set E
62
+ # Need to +1 in range to include the end-point.
63
+ src_phrase = " ".join(srctext[e_start:e_end+1])
64
+ trg_phrase = " ".join(trgtext[fs:fe+1])
65
+ # Include more data for later ordering.
66
+ phrases.add(((e_start, e_end+1), (f_start, f_end+1),
67
+ src_phrase, trg_phrase))
68
+ fe += 1
69
+ if fe in f_aligned or fe == trglen:
70
+ break
71
+ fs -=1
72
+ if fs in f_aligned or fs < 0:
73
+ break
74
+ return phrases
75
+
76
+ def phrase_extraction(srctext, trgtext, alignment, max_phrase_length=0):
77
+ """
78
+ Phrase extraction algorithm extracts all consistent phrase pairs from
79
+ a word-aligned sentence pair.
80
+
81
+ The idea is to loop over all possible source language (e) phrases and find
82
+ the minimal foreign phrase (f) that matches each of them. Matching is done
83
+ by identifying all alignment points for the source phrase and finding the
84
+ shortest foreign phrase that includes all the foreign counterparts for the
85
+ source words.
86
+
87
+ In short, a phrase alignment has to
88
+ (a) contain all alignment points for all covered words
89
+ (b) contain at least one alignment point
90
+
91
+ >>> srctext = "michael assumes that he will stay in the house"
92
+ >>> trgtext = "michael geht davon aus , dass er im haus bleibt"
93
+ >>> alignment = [(0,0), (1,1), (1,2), (1,3), (2,5), (3,6), (4,9),
94
+ ... (5,9), (6,7), (7,7), (8,8)]
95
+ >>> phrases = phrase_extraction(srctext, trgtext, alignment)
96
+ >>> for i in sorted(phrases):
97
+ ... print(i)
98
+ ...
99
+ ((0, 1), (0, 1), 'michael', 'michael')
100
+ ((0, 2), (0, 4), 'michael assumes', 'michael geht davon aus')
101
+ ((0, 2), (0, 4), 'michael assumes', 'michael geht davon aus ,')
102
+ ((0, 3), (0, 6), 'michael assumes that', 'michael geht davon aus , dass')
103
+ ((0, 4), (0, 7), 'michael assumes that he', 'michael geht davon aus , dass er')
104
+ ((0, 9), (0, 10), 'michael assumes that he will stay in the house', 'michael geht davon aus , dass er im haus bleibt')
105
+ ((1, 2), (1, 4), 'assumes', 'geht davon aus')
106
+ ((1, 2), (1, 4), 'assumes', 'geht davon aus ,')
107
+ ((1, 3), (1, 6), 'assumes that', 'geht davon aus , dass')
108
+ ((1, 4), (1, 7), 'assumes that he', 'geht davon aus , dass er')
109
+ ((1, 9), (1, 10), 'assumes that he will stay in the house', 'geht davon aus , dass er im haus bleibt')
110
+ ((2, 3), (5, 6), 'that', ', dass')
111
+ ((2, 3), (5, 6), 'that', 'dass')
112
+ ((2, 4), (5, 7), 'that he', ', dass er')
113
+ ((2, 4), (5, 7), 'that he', 'dass er')
114
+ ((2, 9), (5, 10), 'that he will stay in the house', ', dass er im haus bleibt')
115
+ ((2, 9), (5, 10), 'that he will stay in the house', 'dass er im haus bleibt')
116
+ ((3, 4), (6, 7), 'he', 'er')
117
+ ((3, 9), (6, 10), 'he will stay in the house', 'er im haus bleibt')
118
+ ((4, 6), (9, 10), 'will stay', 'bleibt')
119
+ ((4, 9), (7, 10), 'will stay in the house', 'im haus bleibt')
120
+ ((6, 8), (7, 8), 'in the', 'im')
121
+ ((6, 9), (7, 9), 'in the house', 'im haus')
122
+ ((8, 9), (8, 9), 'house', 'haus')
123
+
124
+ :type srctext: str
125
+ :param srctext: The sentence string from the source language.
126
+ :type trgtext: str
127
+ :param trgtext: The sentence string from the target language.
128
+ :type alignment: str
129
+ :param alignment: The word alignment outputs as list of tuples, where
130
+ the first elements of tuples are the source words' indices and
131
+ second elements are the target words' indices. This is also the output
132
+ format of nltk.translate.ibm1
133
+ :rtype: list(tuple)
134
+ :return: A list of tuples, each element in a list is a phrase and each
135
+ phrase is a tuple made up of (i) its source location, (ii) its target
136
+ location, (iii) the source phrase and (iii) the target phrase. The phrase
137
+ list of tuples represents all the possible phrases extracted from the
138
+ word alignments.
139
+ :type max_phrase_length: int
140
+ :param max_phrase_length: maximal phrase length, if 0 or not specified
141
+ it is set to a length of the longer sentence (srctext or trgtext).
142
+ """
143
+
144
+ srctext = srctext.split() # e
145
+ trgtext = trgtext.split() # f
146
+ srclen = len(srctext) # len(e)
147
+ trglen = len(trgtext) # len(f)
148
+ # Keeps an index of which source/target words that are aligned.
149
+ f_aligned = [j for _,j in alignment]
150
+ max_phrase_length = max_phrase_length or max(srclen,trglen)
151
+
152
+ # set of phrase pairs BP
153
+ bp = set()
154
+
155
+ for e_start in range(srclen):
156
+ max_idx = min(srclen, e_start + max_phrase_length)
157
+ for e_end in range(e_start, max_idx):
158
+ # // find the minimally matching foreign phrase
159
+ # (f start , f end ) = ( length(f), 0 )
160
+ # f_start ∈ [0, len(f) - 1]; f_end ∈ [0, len(f) - 1]
161
+ f_start, f_end = trglen-1 , -1 # 0-based indexing
162
+
163
+ for e,f in alignment:
164
+ if e_start <= e <= e_end:
165
+ f_start = min(f, f_start)
166
+ f_end = max(f, f_end)
167
+ # add extract (f start , f end , e start , e end ) to set BP
168
+ phrases = extract(f_start, f_end, e_start, e_end,
169
+ alignment, f_aligned,
170
+ srctext, trgtext, srclen, trglen,
171
+ max_phrase_length)
172
+ if phrases:
173
+ bp.update(phrases)
174
+ return bp
175
+