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,325 +1,325 @@
1
- # -*- coding: utf-8 -*-
2
- # Natural Language Toolkit: RIBES Score
3
- #
4
- # Copyright (C) 2001-2015 NLTK Project
5
- # Contributors: Katsuhito Sudoh, Liling Tan, Kasramvd, J.F.Sebastian
6
- # Mark Byers, ekhumoro, P. Ortiz
7
- # URL: <http://nltk.org/>
8
- # For license information, see LICENSE.TXT
9
- """ RIBES score implementation """
10
- from __future__ import division
11
- from itertools import islice
12
- import math
13
-
14
- from nltk.util import ngrams, choose
15
-
16
-
17
- def sentence_ribes(references, hypothesis, alpha=0.25, beta=0.10):
18
- """
19
- The RIBES (Rank-based Intuitive Bilingual Evaluation Score) from
20
- Hideki Isozaki, Tsutomu Hirao, Kevin Duh, Katsuhito Sudoh and
21
- Hajime Tsukada. 2010. "Automatic Evaluation of Translation Quality for
22
- Distant Language Pairs". In Proceedings of EMNLP.
23
- http://www.aclweb.org/anthology/D/D10/D10-1092.pdf
24
-
25
- The generic RIBES scores used in shared task, e.g. Workshop for
26
- Asian Translation (WAT) uses the following RIBES calculations:
27
-
28
- RIBES = kendall_tau * (alpha**p1) * (beta**bp)
29
-
30
- Please note that this re-implementation differs from the official
31
- RIBES implementation and though it emulates the results as describe
32
- in the original paper, there are further optimization implemented
33
- in the official RIBES script.
34
-
35
- Users are encouraged to use the official RIBES script instead of this
36
- implementation when evaluating your machine translation system. Refer
37
- to http://www.kecl.ntt.co.jp/icl/lirg/ribes/ for the official script.
38
-
39
- :param references: a list of reference sentences
40
- :type reference: list(list(str))
41
- :param hypothesis: a hypothesis sentence
42
- :type hypothesis: list(str)
43
- :param alpha: hyperparameter used as a prior for the unigram precision.
44
- :type alpha: float
45
- :param beta: hyperparameter used as a prior for the brevity penalty.
46
- :type beta: float
47
- :return: The best ribes score from one of the references.
48
- :rtype: float
49
- """
50
- best_ribes = -1.0
51
- # Calculates RIBES for each reference and returns the best score.
52
- for reference in references:
53
- # Collects the *worder* from the ranked correlation alignments.
54
- worder = word_rank_alignment(reference, hypothesis)
55
- nkt = kendall_tau(worder)
56
-
57
- # Calculates the brevity penalty
58
- bp = min(1.0, math.exp(1.0 - len(reference)/len(hypothesis)))
59
-
60
- # Calculates the unigram precision, *p1*
61
- p1 = len(worder) / len(hypothesis)
62
-
63
- _ribes = nkt * (p1 ** alpha) * (bp ** beta)
64
-
65
- if _ribes > best_ribes: # Keeps the best score.
66
- best_ribes = _ribes
67
-
68
- return best_ribes
69
-
70
-
71
- def corpus_ribes(list_of_references, hypotheses, alpha=0.25, beta=0.10):
72
- """
73
- This function "calculates RIBES for a system output (hypothesis) with
74
- multiple references, and returns "best" score among multi-references and
75
- individual scores. The scores are corpus-wise, i.e., averaged by the number
76
- of sentences." (c.f. RIBES version 1.03.1 code).
77
-
78
- Different from BLEU's micro-average precision, RIBES calculates the
79
- macro-average precision by averaging the best RIBES score for each pair of
80
- hypothesis and its corresponding references
81
-
82
- >>> hyp1 = ['It', 'is', 'a', 'guide', 'to', 'action', 'which',
83
- ... 'ensures', 'that', 'the', 'military', 'always',
84
- ... 'obeys', 'the', 'commands', 'of', 'the', 'party']
85
- >>> ref1a = ['It', 'is', 'a', 'guide', 'to', 'action', 'that',
86
- ... 'ensures', 'that', 'the', 'military', 'will', 'forever',
87
- ... 'heed', 'Party', 'commands']
88
- >>> ref1b = ['It', 'is', 'the', 'guiding', 'principle', 'which',
89
- ... 'guarantees', 'the', 'military', 'forces', 'always',
90
- ... 'being', 'under', 'the', 'command', 'of', 'the', 'Party']
91
- >>> ref1c = ['It', 'is', 'the', 'practical', 'guide', 'for', 'the',
92
- ... 'army', 'always', 'to', 'heed', 'the', 'directions',
93
- ... 'of', 'the', 'party']
94
-
95
- >>> hyp2 = ['he', 'read', 'the', 'book', 'because', 'he', 'was',
96
- ... 'interested', 'in', 'world', 'history']
97
- >>> ref2a = ['he', 'was', 'interested', 'in', 'world', 'history',
98
- ... 'because', 'he', 'read', 'the', 'book']
99
-
100
- >>> list_of_references = [[ref1a, ref1b, ref1c], [ref2a]]
101
- >>> hypotheses = [hyp1, hyp2]
102
- >>> round(corpus_ribes(list_of_references, hypotheses),4)
103
- 0.3597
104
-
105
- :param references: a corpus of lists of reference sentences, w.r.t. hypotheses
106
- :type references: list(list(list(str)))
107
- :param hypotheses: a list of hypothesis sentences
108
- :type hypotheses: list(list(str))
109
- :param alpha: hyperparameter used as a prior for the unigram precision.
110
- :type alpha: float
111
- :param beta: hyperparameter used as a prior for the brevity penalty.
112
- :type beta: float
113
- :return: The best ribes score from one of the references.
114
- :rtype: float
115
- """
116
- corpus_best_ribes = 0.0
117
- # Iterate through each hypothesis and their corresponding references.
118
- for references, hypothesis in zip(list_of_references, hypotheses):
119
- corpus_best_ribes += sentence_ribes(references, hypothesis, alpha, beta)
120
- return corpus_best_ribes / len(hypotheses)
121
-
122
-
123
- def position_of_ngram(ngram, sentence):
124
- """
125
- This function returns the position of the first instance of the ngram
126
- appearing in a sentence.
127
-
128
- Note that one could also use string as follows but the code is a little
129
- convoluted with type casting back and forth:
130
-
131
- char_pos = ' '.join(sent)[:' '.join(sent).index(' '.join(ngram))]
132
- word_pos = char_pos.count(' ')
133
-
134
- Another way to conceive this is:
135
-
136
- return next(i for i, ng in enumerate(ngrams(sentence, len(ngram)))
137
- if ng == ngram)
138
-
139
- :param ngram: The ngram that needs to be searched
140
- :type ngram: tuple
141
- :param sentence: The list of tokens to search from.
142
- :type sentence: list(str)
143
- """
144
- # Iterates through the ngrams in sentence.
145
- for i,sublist in enumerate(ngrams(sentence, len(ngram))):
146
- # Returns the index of the word when ngram matches.
147
- if ngram == sublist:
148
- return i
149
-
150
-
151
- def word_rank_alignment(reference, hypothesis, character_based=False):
152
- """
153
- This is the word rank alignment algorithm described in the paper to produce
154
- the *worder* list, i.e. a list of word indices of the hypothesis word orders
155
- w.r.t. the list of reference words.
156
-
157
- Below is (H0, R0) example from the Isozaki et al. 2010 paper,
158
- note the examples are indexed from 1 but the results here are indexed from 0:
159
-
160
- >>> ref = str('he was interested in world history because he '
161
- ... 'read the book').split()
162
- >>> hyp = str('he read the book because he was interested in world '
163
- ... 'history').split()
164
- >>> word_rank_alignment(ref, hyp)
165
- [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
166
-
167
- The (H1, R1) example from the paper, note the 0th index:
168
-
169
- >>> ref = 'John hit Bob yesterday'.split()
170
- >>> hyp = 'Bob hit John yesterday'.split()
171
- >>> word_rank_alignment(ref, hyp)
172
- [2, 1, 0, 3]
173
-
174
- Here is the (H2, R2) example from the paper, note the 0th index here too:
175
-
176
- >>> ref = 'the boy read the book'.split()
177
- >>> hyp = 'the book was read by the boy'.split()
178
- >>> word_rank_alignment(ref, hyp)
179
- [3, 4, 2, 0, 1]
180
-
181
- :param reference: a reference sentence
182
- :type reference: list(str)
183
- :param hypothesis: a hypothesis sentence
184
- :type hypothesis: list(str)
185
- """
186
- worder = []
187
- hyp_len = len(hypothesis)
188
- # Stores a list of possible ngrams from the reference sentence.
189
- # This is used for matching context window later in the algorithm.
190
- ref_ngrams = []
191
- hyp_ngrams = []
192
- for n in range(1, len(reference)+1):
193
- for ng in ngrams(reference, n):
194
- ref_ngrams.append(ng)
195
- for ng in ngrams(hypothesis, n):
196
- hyp_ngrams.append(ng)
197
- for i, h_word in enumerate(hypothesis):
198
- # If word is not in the reference, continue.
199
- if h_word not in reference:
200
- continue
201
- # If we can determine one-to-one word correspondence for unigrams that
202
- # only appear once in both the reference and hypothesis.
203
- elif hypothesis.count(h_word) == reference.count(h_word) == 1:
204
- worder.append(reference.index(h_word))
205
- else:
206
- max_window_size = max(i, hyp_len-i+1)
207
- for window in range(1, max_window_size):
208
- if i+window < hyp_len: # If searching the right context is possible.
209
- # Retrieve the right context window.
210
- right_context_ngram = tuple(islice(hypothesis, i, i+window+1))
211
- num_times_in_ref = ref_ngrams.count(right_context_ngram)
212
- num_times_in_hyp = hyp_ngrams.count(right_context_ngram)
213
- # If ngram appears only once in both ref and hyp.
214
- if num_times_in_ref == num_times_in_hyp == 1:
215
- # Find the position of ngram that matched the reference.
216
- pos = position_of_ngram(right_context_ngram, reference)
217
- worder.append(pos) # Add the positions of the ngram.
218
- break
219
- if window <= i: # If searching the left context is possible.
220
- # Retrieve the left context window.
221
- left_context_ngram = tuple(islice(hypothesis, i-window, i+1))
222
- num_times_in_ref = ref_ngrams.count(left_context_ngram)
223
- num_times_in_hyp = hyp_ngrams.count(left_context_ngram)
224
- if num_times_in_ref == num_times_in_hyp == 1:
225
- # Find the position of ngram that matched the reference.
226
- pos = position_of_ngram(left_context_ngram, reference)
227
- # Add the positions of the ngram.
228
- worder.append(pos+ len(left_context_ngram) -1)
229
- break
230
- return worder
231
-
232
-
233
- def find_increasing_sequences(worder):
234
- """
235
- Given the *worder* list, this function groups monotonic +1 sequences.
236
-
237
- >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
238
- >>> list(find_increasing_sequences(worder))
239
- [(7, 8, 9, 10), (0, 1, 2, 3, 4, 5)]
240
-
241
- :param worder: The worder list output from word_rank_alignment
242
- :param type: list(int)
243
- """
244
- items = iter(worder)
245
- a, b = None, next(items, None)
246
- result = [b]
247
- while b is not None:
248
- a, b = b, next(items, None)
249
- if b is not None and a + 1 == b:
250
- result.append(b)
251
- else:
252
- if len(result) > 1:
253
- yield tuple(result)
254
- result = [b]
255
-
256
-
257
- def kendall_tau(worder, normalize=True):
258
- """
259
- Calculates the Kendall's Tau correlation coefficient given the *worder*
260
- list of word alignments from word_rank_alignment(), using the formula:
261
-
262
- tau = 2 * num_increasing_pairs / num_possible pairs -1
263
-
264
- Note that the no. of increasing pairs can be discontinuous in the *worder*
265
- list and each each increasing sequence can be tabulated as choose(len(seq), 2)
266
- no. of increasing pairs, e.g.
267
-
268
- >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
269
- >>> number_possible_pairs = choose(len(worder), 2)
270
- >>> round(kendall_tau(worder, normalize=False),3)
271
- -0.236
272
- >>> round(kendall_tau(worder),3)
273
- 0.382
274
-
275
- :param worder: The worder list output from word_rank_alignment
276
- :type worder: list(int)
277
- :param normalize: Flag to indicate normalization
278
- :type normalize: boolean
279
- :return: The Kendall's Tau correlation coefficient.
280
- :rtype: float
281
- """
282
- worder_len = len(worder)
283
- # Extract the groups of increasing/monotonic sequences.
284
- increasing_sequences = find_increasing_sequences(worder)
285
- # Calculate no. of increasing_pairs in *worder* list.
286
- num_increasing_pairs = sum(choose(len(seq),2) for seq in increasing_sequences)
287
- # Calculate no. of possible pairs.
288
- num_possible_pairs = choose(worder_len, 2)
289
- # Kendall's Tau computation.
290
- tau = 2 * num_increasing_pairs / num_possible_pairs -1
291
- if normalize: # If normalized, the tau output falls between 0.0 to 1.0
292
- return (tau + 1) /2
293
- else: # Otherwise, the tau outputs falls between -1.0 to +1.0
294
- return tau
295
-
296
-
297
- def spearman_rho(worder, normalize=True):
298
- """
299
- Calculates the Spearman's Rho correlation coefficient given the *worder*
300
- list of word alignment from word_rank_alignment(), using the formula:
301
-
302
- rho = 1 - sum(d**2) / choose(len(worder)+1, 3)
303
-
304
- Given that d is the sum of difference between the *worder* list of indices
305
- and the original word indices from the reference sentence.
306
-
307
- Using the (H0,R0) and (H5, R5) example from the paper
308
-
309
- >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
310
- >>> round(spearman_rho(worder, normalize=False), 3)
311
- -0.591
312
- >>> round(spearman_rho(worder), 3)
313
- 0.205
314
-
315
- :param worder: The worder list output from word_rank_alignment
316
- :param type: list(int)
317
- """
318
- worder_len = len(worder)
319
- sum_d_square = sum((wi - i)**2 for wi, i in zip(worder, range(worder_len)))
320
- rho = 1 - sum_d_square / choose(worder_len+1, 3)
321
-
322
- if normalize: # If normalized, the rho output falls between 0.0 to 1.0
323
- return (rho + 1) /2
324
- else: # Otherwise, the rho outputs falls between -1.0 to +1.0
325
- return rho
1
+ # -*- coding: utf-8 -*-
2
+ # Natural Language Toolkit: RIBES Score
3
+ #
4
+ # Copyright (C) 2001-2018 NLTK Project
5
+ # Contributors: Katsuhito Sudoh, Liling Tan, Kasramvd, J.F.Sebastian
6
+ # Mark Byers, ekhumoro, P. Ortiz
7
+ # URL: <http://nltk.org/>
8
+ # For license information, see LICENSE.TXT
9
+ """ RIBES score implementation """
10
+ from __future__ import division
11
+ from itertools import islice
12
+ import math
13
+
14
+ from nltk.util import ngrams, choose
15
+
16
+
17
+ def sentence_ribes(references, hypothesis, alpha=0.25, beta=0.10):
18
+ """
19
+ The RIBES (Rank-based Intuitive Bilingual Evaluation Score) from
20
+ Hideki Isozaki, Tsutomu Hirao, Kevin Duh, Katsuhito Sudoh and
21
+ Hajime Tsukada. 2010. "Automatic Evaluation of Translation Quality for
22
+ Distant Language Pairs". In Proceedings of EMNLP.
23
+ http://www.aclweb.org/anthology/D/D10/D10-1092.pdf
24
+
25
+ The generic RIBES scores used in shared task, e.g. Workshop for
26
+ Asian Translation (WAT) uses the following RIBES calculations:
27
+
28
+ RIBES = kendall_tau * (alpha**p1) * (beta**bp)
29
+
30
+ Please note that this re-implementation differs from the official
31
+ RIBES implementation and though it emulates the results as describe
32
+ in the original paper, there are further optimization implemented
33
+ in the official RIBES script.
34
+
35
+ Users are encouraged to use the official RIBES script instead of this
36
+ implementation when evaluating your machine translation system. Refer
37
+ to http://www.kecl.ntt.co.jp/icl/lirg/ribes/ for the official script.
38
+
39
+ :param references: a list of reference sentences
40
+ :type reference: list(list(str))
41
+ :param hypothesis: a hypothesis sentence
42
+ :type hypothesis: list(str)
43
+ :param alpha: hyperparameter used as a prior for the unigram precision.
44
+ :type alpha: float
45
+ :param beta: hyperparameter used as a prior for the brevity penalty.
46
+ :type beta: float
47
+ :return: The best ribes score from one of the references.
48
+ :rtype: float
49
+ """
50
+ best_ribes = -1.0
51
+ # Calculates RIBES for each reference and returns the best score.
52
+ for reference in references:
53
+ # Collects the *worder* from the ranked correlation alignments.
54
+ worder = word_rank_alignment(reference, hypothesis)
55
+ nkt = kendall_tau(worder)
56
+
57
+ # Calculates the brevity penalty
58
+ bp = min(1.0, math.exp(1.0 - len(reference)/len(hypothesis)))
59
+
60
+ # Calculates the unigram precision, *p1*
61
+ p1 = len(worder) / len(hypothesis)
62
+
63
+ _ribes = nkt * (p1 ** alpha) * (bp ** beta)
64
+
65
+ if _ribes > best_ribes: # Keeps the best score.
66
+ best_ribes = _ribes
67
+
68
+ return best_ribes
69
+
70
+
71
+ def corpus_ribes(list_of_references, hypotheses, alpha=0.25, beta=0.10):
72
+ """
73
+ This function "calculates RIBES for a system output (hypothesis) with
74
+ multiple references, and returns "best" score among multi-references and
75
+ individual scores. The scores are corpus-wise, i.e., averaged by the number
76
+ of sentences." (c.f. RIBES version 1.03.1 code).
77
+
78
+ Different from BLEU's micro-average precision, RIBES calculates the
79
+ macro-average precision by averaging the best RIBES score for each pair of
80
+ hypothesis and its corresponding references
81
+
82
+ >>> hyp1 = ['It', 'is', 'a', 'guide', 'to', 'action', 'which',
83
+ ... 'ensures', 'that', 'the', 'military', 'always',
84
+ ... 'obeys', 'the', 'commands', 'of', 'the', 'party']
85
+ >>> ref1a = ['It', 'is', 'a', 'guide', 'to', 'action', 'that',
86
+ ... 'ensures', 'that', 'the', 'military', 'will', 'forever',
87
+ ... 'heed', 'Party', 'commands']
88
+ >>> ref1b = ['It', 'is', 'the', 'guiding', 'principle', 'which',
89
+ ... 'guarantees', 'the', 'military', 'forces', 'always',
90
+ ... 'being', 'under', 'the', 'command', 'of', 'the', 'Party']
91
+ >>> ref1c = ['It', 'is', 'the', 'practical', 'guide', 'for', 'the',
92
+ ... 'army', 'always', 'to', 'heed', 'the', 'directions',
93
+ ... 'of', 'the', 'party']
94
+
95
+ >>> hyp2 = ['he', 'read', 'the', 'book', 'because', 'he', 'was',
96
+ ... 'interested', 'in', 'world', 'history']
97
+ >>> ref2a = ['he', 'was', 'interested', 'in', 'world', 'history',
98
+ ... 'because', 'he', 'read', 'the', 'book']
99
+
100
+ >>> list_of_references = [[ref1a, ref1b, ref1c], [ref2a]]
101
+ >>> hypotheses = [hyp1, hyp2]
102
+ >>> round(corpus_ribes(list_of_references, hypotheses),4)
103
+ 0.3597
104
+
105
+ :param references: a corpus of lists of reference sentences, w.r.t. hypotheses
106
+ :type references: list(list(list(str)))
107
+ :param hypotheses: a list of hypothesis sentences
108
+ :type hypotheses: list(list(str))
109
+ :param alpha: hyperparameter used as a prior for the unigram precision.
110
+ :type alpha: float
111
+ :param beta: hyperparameter used as a prior for the brevity penalty.
112
+ :type beta: float
113
+ :return: The best ribes score from one of the references.
114
+ :rtype: float
115
+ """
116
+ corpus_best_ribes = 0.0
117
+ # Iterate through each hypothesis and their corresponding references.
118
+ for references, hypothesis in zip(list_of_references, hypotheses):
119
+ corpus_best_ribes += sentence_ribes(references, hypothesis, alpha, beta)
120
+ return corpus_best_ribes / len(hypotheses)
121
+
122
+
123
+ def position_of_ngram(ngram, sentence):
124
+ """
125
+ This function returns the position of the first instance of the ngram
126
+ appearing in a sentence.
127
+
128
+ Note that one could also use string as follows but the code is a little
129
+ convoluted with type casting back and forth:
130
+
131
+ char_pos = ' '.join(sent)[:' '.join(sent).index(' '.join(ngram))]
132
+ word_pos = char_pos.count(' ')
133
+
134
+ Another way to conceive this is:
135
+
136
+ return next(i for i, ng in enumerate(ngrams(sentence, len(ngram)))
137
+ if ng == ngram)
138
+
139
+ :param ngram: The ngram that needs to be searched
140
+ :type ngram: tuple
141
+ :param sentence: The list of tokens to search from.
142
+ :type sentence: list(str)
143
+ """
144
+ # Iterates through the ngrams in sentence.
145
+ for i,sublist in enumerate(ngrams(sentence, len(ngram))):
146
+ # Returns the index of the word when ngram matches.
147
+ if ngram == sublist:
148
+ return i
149
+
150
+
151
+ def word_rank_alignment(reference, hypothesis, character_based=False):
152
+ """
153
+ This is the word rank alignment algorithm described in the paper to produce
154
+ the *worder* list, i.e. a list of word indices of the hypothesis word orders
155
+ w.r.t. the list of reference words.
156
+
157
+ Below is (H0, R0) example from the Isozaki et al. 2010 paper,
158
+ note the examples are indexed from 1 but the results here are indexed from 0:
159
+
160
+ >>> ref = str('he was interested in world history because he '
161
+ ... 'read the book').split()
162
+ >>> hyp = str('he read the book because he was interested in world '
163
+ ... 'history').split()
164
+ >>> word_rank_alignment(ref, hyp)
165
+ [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
166
+
167
+ The (H1, R1) example from the paper, note the 0th index:
168
+
169
+ >>> ref = 'John hit Bob yesterday'.split()
170
+ >>> hyp = 'Bob hit John yesterday'.split()
171
+ >>> word_rank_alignment(ref, hyp)
172
+ [2, 1, 0, 3]
173
+
174
+ Here is the (H2, R2) example from the paper, note the 0th index here too:
175
+
176
+ >>> ref = 'the boy read the book'.split()
177
+ >>> hyp = 'the book was read by the boy'.split()
178
+ >>> word_rank_alignment(ref, hyp)
179
+ [3, 4, 2, 0, 1]
180
+
181
+ :param reference: a reference sentence
182
+ :type reference: list(str)
183
+ :param hypothesis: a hypothesis sentence
184
+ :type hypothesis: list(str)
185
+ """
186
+ worder = []
187
+ hyp_len = len(hypothesis)
188
+ # Stores a list of possible ngrams from the reference sentence.
189
+ # This is used for matching context window later in the algorithm.
190
+ ref_ngrams = []
191
+ hyp_ngrams = []
192
+ for n in range(1, len(reference)+1):
193
+ for ng in ngrams(reference, n):
194
+ ref_ngrams.append(ng)
195
+ for ng in ngrams(hypothesis, n):
196
+ hyp_ngrams.append(ng)
197
+ for i, h_word in enumerate(hypothesis):
198
+ # If word is not in the reference, continue.
199
+ if h_word not in reference:
200
+ continue
201
+ # If we can determine one-to-one word correspondence for unigrams that
202
+ # only appear once in both the reference and hypothesis.
203
+ elif hypothesis.count(h_word) == reference.count(h_word) == 1:
204
+ worder.append(reference.index(h_word))
205
+ else:
206
+ max_window_size = max(i, hyp_len-i+1)
207
+ for window in range(1, max_window_size):
208
+ if i+window < hyp_len: # If searching the right context is possible.
209
+ # Retrieve the right context window.
210
+ right_context_ngram = tuple(islice(hypothesis, i, i+window+1))
211
+ num_times_in_ref = ref_ngrams.count(right_context_ngram)
212
+ num_times_in_hyp = hyp_ngrams.count(right_context_ngram)
213
+ # If ngram appears only once in both ref and hyp.
214
+ if num_times_in_ref == num_times_in_hyp == 1:
215
+ # Find the position of ngram that matched the reference.
216
+ pos = position_of_ngram(right_context_ngram, reference)
217
+ worder.append(pos) # Add the positions of the ngram.
218
+ break
219
+ if window <= i: # If searching the left context is possible.
220
+ # Retrieve the left context window.
221
+ left_context_ngram = tuple(islice(hypothesis, i-window, i+1))
222
+ num_times_in_ref = ref_ngrams.count(left_context_ngram)
223
+ num_times_in_hyp = hyp_ngrams.count(left_context_ngram)
224
+ if num_times_in_ref == num_times_in_hyp == 1:
225
+ # Find the position of ngram that matched the reference.
226
+ pos = position_of_ngram(left_context_ngram, reference)
227
+ # Add the positions of the ngram.
228
+ worder.append(pos+ len(left_context_ngram) -1)
229
+ break
230
+ return worder
231
+
232
+
233
+ def find_increasing_sequences(worder):
234
+ """
235
+ Given the *worder* list, this function groups monotonic +1 sequences.
236
+
237
+ >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
238
+ >>> list(find_increasing_sequences(worder))
239
+ [(7, 8, 9, 10), (0, 1, 2, 3, 4, 5)]
240
+
241
+ :param worder: The worder list output from word_rank_alignment
242
+ :param type: list(int)
243
+ """
244
+ items = iter(worder)
245
+ a, b = None, next(items, None)
246
+ result = [b]
247
+ while b is not None:
248
+ a, b = b, next(items, None)
249
+ if b is not None and a + 1 == b:
250
+ result.append(b)
251
+ else:
252
+ if len(result) > 1:
253
+ yield tuple(result)
254
+ result = [b]
255
+
256
+
257
+ def kendall_tau(worder, normalize=True):
258
+ """
259
+ Calculates the Kendall's Tau correlation coefficient given the *worder*
260
+ list of word alignments from word_rank_alignment(), using the formula:
261
+
262
+ tau = 2 * num_increasing_pairs / num_possible pairs -1
263
+
264
+ Note that the no. of increasing pairs can be discontinuous in the *worder*
265
+ list and each each increasing sequence can be tabulated as choose(len(seq), 2)
266
+ no. of increasing pairs, e.g.
267
+
268
+ >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
269
+ >>> number_possible_pairs = choose(len(worder), 2)
270
+ >>> round(kendall_tau(worder, normalize=False),3)
271
+ -0.236
272
+ >>> round(kendall_tau(worder),3)
273
+ 0.382
274
+
275
+ :param worder: The worder list output from word_rank_alignment
276
+ :type worder: list(int)
277
+ :param normalize: Flag to indicate normalization
278
+ :type normalize: boolean
279
+ :return: The Kendall's Tau correlation coefficient.
280
+ :rtype: float
281
+ """
282
+ worder_len = len(worder)
283
+ # Extract the groups of increasing/monotonic sequences.
284
+ increasing_sequences = find_increasing_sequences(worder)
285
+ # Calculate no. of increasing_pairs in *worder* list.
286
+ num_increasing_pairs = sum(choose(len(seq),2) for seq in increasing_sequences)
287
+ # Calculate no. of possible pairs.
288
+ num_possible_pairs = choose(worder_len, 2)
289
+ # Kendall's Tau computation.
290
+ tau = 2 * num_increasing_pairs / num_possible_pairs -1
291
+ if normalize: # If normalized, the tau output falls between 0.0 to 1.0
292
+ return (tau + 1) /2
293
+ else: # Otherwise, the tau outputs falls between -1.0 to +1.0
294
+ return tau
295
+
296
+
297
+ def spearman_rho(worder, normalize=True):
298
+ """
299
+ Calculates the Spearman's Rho correlation coefficient given the *worder*
300
+ list of word alignment from word_rank_alignment(), using the formula:
301
+
302
+ rho = 1 - sum(d**2) / choose(len(worder)+1, 3)
303
+
304
+ Given that d is the sum of difference between the *worder* list of indices
305
+ and the original word indices from the reference sentence.
306
+
307
+ Using the (H0,R0) and (H5, R5) example from the paper
308
+
309
+ >>> worder = [7, 8, 9, 10, 6, 0, 1, 2, 3, 4, 5]
310
+ >>> round(spearman_rho(worder, normalize=False), 3)
311
+ -0.591
312
+ >>> round(spearman_rho(worder), 3)
313
+ 0.205
314
+
315
+ :param worder: The worder list output from word_rank_alignment
316
+ :param type: list(int)
317
+ """
318
+ worder_len = len(worder)
319
+ sum_d_square = sum((wi - i)**2 for wi, i in zip(worder, range(worder_len)))
320
+ rho = 1 - sum_d_square / choose(worder_len+1, 3)
321
+
322
+ if normalize: # If normalized, the rho output falls between 0.0 to 1.0
323
+ return (rho + 1) /2
324
+ else: # Otherwise, the rho outputs falls between -1.0 to +1.0
325
+ return rho