3.4 BM25算法简介与应用.html 47 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753
  1. <!doctype html>
  2. <html lang="zh" class="no-js">
  3. <head>
  4. <meta charset="utf-8">
  5. <meta name="viewport" content="width=device-width,initial-scale=1">
  6. <link rel="prev" href="3.3%20%E5%9F%BA%E4%BA%8EMysql%E6%95%B0%E6%8D%AE%E5%BA%93%E5%AE%9E%E7%8E%B0%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F.html">
  7. <link rel="next" href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.1%20%E6%95%B4%E4%BD%93%E6%9E%B6%E6%9E%84%E4%B8%8E%E5%B7%A5%E7%A8%8B%E6%B5%81%E7%A8%8B.html">
  8. <link rel="icon" href="../img/AI.jpg">
  9. <meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.21">
  10. <title>3.4 BM25算法简介与应用 - 企业知识库RAG智能问答项目V1.0</title>
  11. <link rel="stylesheet" href="../assets/stylesheets/main.2a3383ac.min.css">
  12. <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
  13. <link rel="stylesheet" href="https://fonts.googleapis.com/css?family=Roboto:300,300i,400,400i,700,700i%7CRoboto+Mono:400,400i,700,700i&amp;display=fallback">
  14. <style>:root{--md-text-font:"Roboto";--md-code-font:"Roboto Mono"}</style>
  15. <link rel="stylesheet" href="../stylesheets/extra.css">
  16. <script>__md_scope=new URL("..",location),__md_hash=e=>[...e].reduce(((e,_)=>(e<<5)-e+_.charCodeAt(0)),0),__md_get=(e,_=localStorage,t=__md_scope)=>JSON.parse(_.getItem(t.pathname+"."+e)),__md_set=(e,_,t=localStorage,a=__md_scope)=>{try{t.setItem(a.pathname+"."+e,JSON.stringify(_))}catch(e){}}</script>
  17. </head>
  18. <body dir="ltr">
  19. <input class="md-toggle" data-md-toggle="drawer" type="checkbox" id="__drawer" autocomplete="off">
  20. <input class="md-toggle" data-md-toggle="search" type="checkbox" id="__search" autocomplete="off">
  21. <label class="md-overlay" for="__drawer"></label>
  22. <div data-md-component="skip">
  23. <a href="#_1" class="md-skip">
  24. 跳转至
  25. </a>
  26. </div>
  27. <div data-md-component="announce">
  28. </div>
  29. <header class="md-header md-header--shadow" data-md-component="header">
  30. <nav class="md-header__inner md-grid" aria-label="页眉">
  31. <a href="../index.html" title="企业知识库RAG智能问答项目V1.0" class="md-header__button md-logo" aria-label="企业知识库RAG智能问答项目V1.0" data-md-component="logo">
  32. <img src="../img/AI.jpg" alt="logo">
  33. </a>
  34. <label class="md-header__button md-icon" for="__drawer">
  35. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M3 6h18v2H3zm0 5h18v2H3zm0 5h18v2H3z"></path></svg>
  36. </label>
  37. <div class="md-header__title" data-md-component="header-title">
  38. <div class="md-header__ellipsis">
  39. <div class="md-header__topic">
  40. <span class="md-ellipsis">
  41. 企业知识库RAG智能问答项目V1.0
  42. </span>
  43. </div>
  44. <div class="md-header__topic" data-md-component="header-topic">
  45. <span class="md-ellipsis">
  46. 3.4 BM25算法简介与应用
  47. </span>
  48. </div>
  49. </div>
  50. </div>
  51. <script>var palette=__md_get("__palette");if(palette&&palette.color){if("(prefers-color-scheme)"===palette.color.media){var media=matchMedia("(prefers-color-scheme: light)"),input=document.querySelector(media.matches?"[data-md-color-media='(prefers-color-scheme: light)']":"[data-md-color-media='(prefers-color-scheme: dark)']");palette.color.media=input.getAttribute("data-md-color-media"),palette.color.scheme=input.getAttribute("data-md-color-scheme"),palette.color.primary=input.getAttribute("data-md-color-primary"),palette.color.accent=input.getAttribute("data-md-color-accent")}for(var[key,value]of Object.entries(palette.color))document.body.setAttribute("data-md-color-"+key,value)}</script>
  52. <label class="md-header__button md-icon" for="__search">
  53. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M9.5 3A6.5 6.5 0 0 1 16 9.5c0 1.61-.59 3.09-1.56 4.23l.27.27h.79l5 5-1.5 1.5-5-5v-.79l-.27-.27A6.52 6.52 0 0 1 9.5 16 6.5 6.5 0 0 1 3 9.5 6.5 6.5 0 0 1 9.5 3m0 2C7 5 5 7 5 9.5S7 14 9.5 14 14 12 14 9.5 12 5 9.5 5"></path></svg>
  54. </label>
  55. <div class="md-search" data-md-component="search" role="dialog">
  56. <label class="md-search__overlay" for="__search"></label>
  57. <div class="md-search__inner" role="search">
  58. <form class="md-search__form" name="search">
  59. <input type="text" class="md-search__input" name="query" aria-label="搜索" placeholder="搜索" autocapitalize="off" autocorrect="off" autocomplete="off" spellcheck="false" data-md-component="search-query" required>
  60. <label class="md-search__icon md-icon" for="__search">
  61. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M9.5 3A6.5 6.5 0 0 1 16 9.5c0 1.61-.59 3.09-1.56 4.23l.27.27h.79l5 5-1.5 1.5-5-5v-.79l-.27-.27A6.52 6.52 0 0 1 9.5 16 6.5 6.5 0 0 1 3 9.5 6.5 6.5 0 0 1 9.5 3m0 2C7 5 5 7 5 9.5S7 14 9.5 14 14 12 14 9.5 12 5 9.5 5"></path></svg>
  62. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M20 11v2H8l5.5 5.5-1.42 1.42L4.16 12l7.92-7.92L13.5 5.5 8 11z"></path></svg>
  63. </label>
  64. <nav class="md-search__options" aria-label="查找">
  65. <button type="reset" class="md-search__icon md-icon" title="清空当前内容" aria-label="清空当前内容" tabindex="-1">
  66. <svg xmlns="http://www.w3.org/2000/svg" viewbox="0 0 24 24"><path d="M19 6.41 17.59 5 12 10.59 6.41 5 5 6.41 10.59 12 5 17.59 6.41 19 12 13.41 17.59 19 19 17.59 13.41 12z"></path></svg>
  67. </button>
  68. </nav>
  69. </form>
  70. <div class="md-search__output">
  71. <div class="md-search__scrollwrap" tabindex="0" data-md-scrollfix>
  72. <div class="md-search-result" data-md-component="search-result">
  73. <div class="md-search-result__meta">
  74. 正在初始化搜索引擎
  75. </div>
  76. <ol class="md-search-result__list" role="presentation"></ol>
  77. </div>
  78. </div>
  79. </div>
  80. </div>
  81. </div>
  82. </nav>
  83. </header>
  84. <div class="md-container" data-md-component="container">
  85. <main class="md-main" data-md-component="main">
  86. <div class="md-main__inner md-grid">
  87. <div class="md-sidebar md-sidebar--primary" data-md-component="sidebar" data-md-type="navigation">
  88. <div class="md-sidebar__scrollwrap">
  89. <div class="md-sidebar__inner">
  90. <nav class="md-nav md-nav--primary" aria-label="导航栏" data-md-level="0">
  91. <label class="md-nav__title" for="__drawer">
  92. <a href="../index.html" title="企业知识库RAG智能问答项目V1.0" class="md-nav__button md-logo" aria-label="企业知识库RAG智能问答项目V1.0" data-md-component="logo">
  93. <img src="../img/AI.jpg" alt="logo">
  94. </a>
  95. 企业知识库RAG智能问答项目V1.0
  96. </label>
  97. <ul class="md-nav__list" data-md-scrollfix>
  98. <li class="md-nav__item">
  99. <a href="../index.html" class="md-nav__link">
  100. <span class="md-ellipsis">
  101. 知识库内容
  102. </span>
  103. </a>
  104. </li>
  105. <li class="md-nav__item md-nav__item--nested">
  106. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_2">
  107. <label class="md-nav__link" for="__nav_2" id="__nav_2_label" tabindex="0">
  108. <span class="md-ellipsis">
  109. ① 项目概述
  110. </span>
  111. <span class="md-nav__icon md-icon"></span>
  112. </label>
  113. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_2_label" aria-expanded="false">
  114. <label class="md-nav__title" for="__nav_2">
  115. <span class="md-nav__icon md-icon"></span>
  116. ① 项目概述
  117. </label>
  118. <ul class="md-nav__list" data-md-scrollfix>
  119. <li class="md-nav__item">
  120. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.1%20%E9%A1%B9%E7%9B%AE%E6%9E%B6%E6%9E%84.html" class="md-nav__link">
  121. <span class="md-ellipsis">
  122. 1.1 项目架构
  123. </span>
  124. </a>
  125. </li>
  126. <li class="md-nav__item">
  127. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.2%20%E9%A1%B9%E7%9B%AE%E7%8E%AF%E5%A2%83%28window%29.html" class="md-nav__link">
  128. <span class="md-ellipsis">
  129. 1.2 项目环境(window)
  130. </span>
  131. </a>
  132. </li>
  133. <li class="md-nav__item">
  134. <a href="../%E2%91%A0%20%E9%A1%B9%E7%9B%AE%E6%A6%82%E8%BF%B0/1.3%20%E9%A1%B9%E7%9B%AE%E7%8E%AF%E5%A2%83%28mac%29.html" class="md-nav__link">
  135. <span class="md-ellipsis">
  136. 1.3 项目环境(mac)
  137. </span>
  138. </a>
  139. </li>
  140. </ul>
  141. </nav>
  142. </li>
  143. <li class="md-nav__item md-nav__item--nested">
  144. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_3">
  145. <label class="md-nav__link" for="__nav_3" id="__nav_3_label" tabindex="0">
  146. <span class="md-ellipsis">
  147. ② 项目工具
  148. </span>
  149. <span class="md-nav__icon md-icon"></span>
  150. </label>
  151. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_3_label" aria-expanded="false">
  152. <label class="md-nav__title" for="__nav_3">
  153. <span class="md-nav__icon md-icon"></span>
  154. ② 项目工具
  155. </label>
  156. <ul class="md-nav__list" data-md-scrollfix>
  157. <li class="md-nav__item">
  158. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.1%20Ollama%E5%A4%A7%E6%A8%A1%E5%9E%8B%E9%AB%98%E6%95%88%E7%AE%A1%E7%90%86%E5%B7%A5%E5%85%B7.html" class="md-nav__link">
  159. <span class="md-ellipsis">
  160. 2.1 Ollama大模型高效管理工具
  161. </span>
  162. </a>
  163. </li>
  164. <li class="md-nav__item">
  165. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.2%20LangChain%E5%9F%BA%E7%A1%80%E7%9F%A5%E8%AF%86.md.html" class="md-nav__link">
  166. <span class="md-ellipsis">
  167. 2.2 LangChain基础知识.md
  168. </span>
  169. </a>
  170. </li>
  171. <li class="md-nav__item">
  172. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.3%20%E5%90%91%E9%87%8F%E7%B4%A2%E5%BC%95%EF%BC%9AHNSW%E3%80%81IVF%E4%B8%8EIVF_PQ.html" class="md-nav__link">
  173. <span class="md-ellipsis">
  174. 2.3 向量索引:HNSW、IVF与IVF_PQ
  175. </span>
  176. </a>
  177. </li>
  178. <li class="md-nav__item">
  179. <a href="../%E2%91%A1%20%E9%A1%B9%E7%9B%AE%E5%B7%A5%E5%85%B7/2.4%20Milvus%E5%90%91%E9%87%8F%E6%95%B0%E6%8D%AE%E5%BA%93.html" class="md-nav__link">
  180. <span class="md-ellipsis">
  181. 2.4 Milvus向量数据库
  182. </span>
  183. </a>
  184. </li>
  185. </ul>
  186. </nav>
  187. </li>
  188. <li class="md-nav__item md-nav__item--active md-nav__item--nested">
  189. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_4" checked>
  190. <label class="md-nav__link" for="__nav_4" id="__nav_4_label" tabindex="0">
  191. <span class="md-ellipsis">
  192. ③ 基于Mysql库的问答系统
  193. </span>
  194. <span class="md-nav__icon md-icon"></span>
  195. </label>
  196. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_4_label" aria-expanded="true">
  197. <label class="md-nav__title" for="__nav_4">
  198. <span class="md-nav__icon md-icon"></span>
  199. ③ 基于Mysql库的问答系统
  200. </label>
  201. <ul class="md-nav__list" data-md-scrollfix>
  202. <li class="md-nav__item">
  203. <a href="3.1%20Python%E6%97%A5%E5%BF%97%E4%BB%8B%E7%BB%8D%E4%B8%8E%E5%BA%94%E7%94%A8.html" class="md-nav__link">
  204. <span class="md-ellipsis">
  205. 3.1 Python日志介绍与应用
  206. </span>
  207. </a>
  208. </li>
  209. <li class="md-nav__item">
  210. <a href="3.2%20Redis%E6%95%B0%E6%8D%AE%E5%BA%93%E7%AE%80%E4%BB%8B%E4%B8%8E%E4%BD%BF%E7%94%A8.html" class="md-nav__link">
  211. <span class="md-ellipsis">
  212. 3.2 Redis数据库简介与使用
  213. </span>
  214. </a>
  215. </li>
  216. <li class="md-nav__item">
  217. <a href="3.3%20%E5%9F%BA%E4%BA%8EMysql%E6%95%B0%E6%8D%AE%E5%BA%93%E5%AE%9E%E7%8E%B0%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F.html" class="md-nav__link">
  218. <span class="md-ellipsis">
  219. 3.3 基于Mysql数据库实现问答系统
  220. </span>
  221. </a>
  222. </li>
  223. <li class="md-nav__item md-nav__item--active">
  224. <input class="md-nav__toggle md-toggle" type="checkbox" id="__toc">
  225. <label class="md-nav__link md-nav__link--active" for="__toc">
  226. <span class="md-ellipsis">
  227. 3.4 BM25算法简介与应用
  228. </span>
  229. <span class="md-nav__icon md-icon"></span>
  230. </label>
  231. <a href="3.4%20BM25%E7%AE%97%E6%B3%95%E7%AE%80%E4%BB%8B%E4%B8%8E%E5%BA%94%E7%94%A8.html" class="md-nav__link md-nav__link--active">
  232. <span class="md-ellipsis">
  233. 3.4 BM25算法简介与应用
  234. </span>
  235. </a>
  236. <nav class="md-nav md-nav--secondary" aria-label="目录">
  237. <label class="md-nav__title" for="__toc">
  238. <span class="md-nav__icon md-icon"></span>
  239. 目录
  240. </label>
  241. <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix>
  242. <li class="md-nav__item">
  243. <a href="#_1" class="md-nav__link">
  244. <span class="md-ellipsis">
  245. 学习目标
  246. </span>
  247. </a>
  248. </li>
  249. <li class="md-nav__item">
  250. <a href="#bm25" class="md-nav__link">
  251. <span class="md-ellipsis">
  252. 一、BM25算法概述
  253. </span>
  254. </a>
  255. <nav class="md-nav" aria-label="一、BM25算法概述">
  256. <ul class="md-nav__list">
  257. <li class="md-nav__item">
  258. <a href="#1" class="md-nav__link">
  259. <span class="md-ellipsis">
  260. 1 公式
  261. </span>
  262. </a>
  263. </li>
  264. <li class="md-nav__item">
  265. <a href="#2" class="md-nav__link">
  266. <span class="md-ellipsis">
  267. 2 简单示例
  268. </span>
  269. </a>
  270. </li>
  271. </ul>
  272. </nav>
  273. </li>
  274. <li class="md-nav__item">
  275. <a href="#_2" class="md-nav__link">
  276. <span class="md-ellipsis">
  277. 二、代码实现
  278. </span>
  279. </a>
  280. <nav class="md-nav" aria-label="二、代码实现">
  281. <ul class="md-nav__list">
  282. <li class="md-nav__item">
  283. <a href="#1_1" class="md-nav__link">
  284. <span class="md-ellipsis">
  285. 1 整体结构
  286. </span>
  287. </a>
  288. </li>
  289. <li class="md-nav__item">
  290. <a href="#2_1" class="md-nav__link">
  291. <span class="md-ellipsis">
  292. 2 具体模块
  293. </span>
  294. </a>
  295. <nav class="md-nav" aria-label="2 具体模块">
  296. <ul class="md-nav__list">
  297. <li class="md-nav__item">
  298. <a href="#21-retrievalbm25_searchpy" class="md-nav__link">
  299. <span class="md-ellipsis">
  300. 2.1 检索模块 (retrieval/bm25_search.py)
  301. </span>
  302. </a>
  303. </li>
  304. <li class="md-nav__item">
  305. <a href="#22-mainpy" class="md-nav__link">
  306. <span class="md-ellipsis">
  307. 2.2 主程序 (main.py)
  308. </span>
  309. </a>
  310. </li>
  311. <li class="md-nav__item">
  312. <a href="#23-requirementstxt" class="md-nav__link">
  313. <span class="md-ellipsis">
  314. 2.3 依赖文件 (requirements.txt)
  315. </span>
  316. </a>
  317. </li>
  318. </ul>
  319. </nav>
  320. </li>
  321. </ul>
  322. </nav>
  323. </li>
  324. <li class="md-nav__item">
  325. <a href="#_3" class="md-nav__link">
  326. <span class="md-ellipsis">
  327. 三、示例运行结果
  328. </span>
  329. </a>
  330. </li>
  331. <li class="md-nav__item">
  332. <a href="#_4" class="md-nav__link">
  333. <span class="md-ellipsis">
  334. 本章小结
  335. </span>
  336. </a>
  337. </li>
  338. </ul>
  339. </nav>
  340. </li>
  341. </ul>
  342. </nav>
  343. </li>
  344. <li class="md-nav__item md-nav__item--nested">
  345. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_5">
  346. <label class="md-nav__link" for="__nav_5" id="__nav_5_label" tabindex="0">
  347. <span class="md-ellipsis">
  348. ④ 基于Milvus库的问答系统
  349. </span>
  350. <span class="md-nav__icon md-icon"></span>
  351. </label>
  352. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_5_label" aria-expanded="false">
  353. <label class="md-nav__title" for="__nav_5">
  354. <span class="md-nav__icon md-icon"></span>
  355. ④ 基于Milvus库的问答系统
  356. </label>
  357. <ul class="md-nav__list" data-md-scrollfix>
  358. <li class="md-nav__item">
  359. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.1%20%E6%95%B4%E4%BD%93%E6%9E%B6%E6%9E%84%E4%B8%8E%E5%B7%A5%E7%A8%8B%E6%B5%81%E7%A8%8B.html" class="md-nav__link">
  360. <span class="md-ellipsis">
  361. 4.1 整体架构与工程流程
  362. </span>
  363. </a>
  364. </li>
  365. <li class="md-nav__item">
  366. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.2%20%E5%9F%BA%E7%A1%80%E6%A8%A1%E5%9D%97.html" class="md-nav__link">
  367. <span class="md-ellipsis">
  368. 4.2 基础模块
  369. </span>
  370. </a>
  371. </li>
  372. <li class="md-nav__item">
  373. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.3%20%E6%96%87%E6%A1%A3%E5%A4%84%E7%90%86%E6%A8%A1%E5%9D%97.html" class="md-nav__link">
  374. <span class="md-ellipsis">
  375. 4.3 文档处理模块
  376. </span>
  377. </a>
  378. </li>
  379. <li class="md-nav__item">
  380. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.4%20%E5%90%91%E9%87%8F%E5%AD%98%E5%82%A8.html" class="md-nav__link">
  381. <span class="md-ellipsis">
  382. 4.4 向量存储
  383. </span>
  384. </a>
  385. </li>
  386. <li class="md-nav__item"><a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.5%20BERT%E5%9F%BA%E7%A1%80%E4%B8%8E%E4%BB%BB%E5%8A%A1%E5%BE%AE%E8%B0%83.html" class="md-nav__link"><span class="md-ellipsis">4.5 BERT基础与任务微调</span></a></li><li class="md-nav__item">
  387. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.6%20prompts%E8%AE%BE%E8%AE%A1%E4%B8%8Equery%E6%84%8F%E5%9B%BE%E8%AF%86%E5%88%AB.html" class="md-nav__link">
  388. <span class="md-ellipsis">
  389. 4.6 prompts设计与query意图识别
  390. </span>
  391. </a>
  392. </li>
  393. <li class="md-nav__item">
  394. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.7%20%E6%A3%80%E7%B4%A2%E7%AD%96%E7%95%A5%E4%B8%8ERAG%E7%B3%BB%E7%BB%9F%E8%AE%BE%E8%AE%A1.html" class="md-nav__link">
  395. <span class="md-ellipsis">
  396. 4.7 检索策略与RAG系统设计
  397. </span>
  398. </a>
  399. </li>
  400. <li class="md-nav__item">
  401. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.8%20RAG%E7%B3%BB%E7%BB%9F%E8%BF%90%E8%A1%8C.html" class="md-nav__link">
  402. <span class="md-ellipsis">
  403. 4.8 RAG系统运行
  404. </span>
  405. </a>
  406. </li>
  407. <li class="md-nav__item">
  408. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.9%20%E6%96%87%E6%A1%A3%E8%A7%A3%E6%9E%90%E5%B7%A5%E5%85%B7%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  409. <span class="md-ellipsis">
  410. 4.9 文档解析工具(扩展资料)
  411. </span>
  412. </a>
  413. </li>
  414. <li class="md-nav__item">
  415. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/4.10%20RAG%E4%B8%AD%E7%9A%84Query%E6%94%B9%E5%86%99%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  416. <span class="md-ellipsis">
  417. 4.10 RAG中的Query改写(扩展资料)
  418. </span>
  419. </a>
  420. </li>
  421. <li class="md-nav__item">
  422. <a href="../%E2%91%A3%20%E5%9F%BA%E4%BA%8EMilvus%E5%BA%93%E7%9A%84%E9%97%AE%E7%AD%94%E7%B3%BB%E7%BB%9F/%E6%95%B0%E6%8D%AE%E9%9B%86%E7%94%9F%E6%88%90%E4%B8%8E%E4%BC%98%E5%8C%96%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  423. <span class="md-ellipsis">
  424. 数据集生成与优化(扩展资料)
  425. </span>
  426. </a>
  427. </li>
  428. </ul>
  429. </nav>
  430. </li>
  431. <li class="md-nav__item md-nav__item--nested">
  432. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_6">
  433. <label class="md-nav__link" for="__nav_6" id="__nav_6_label" tabindex="0">
  434. <span class="md-ellipsis">
  435. ⑤ RAG系统评估
  436. </span>
  437. <span class="md-nav__icon md-icon"></span>
  438. </label>
  439. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_6_label" aria-expanded="false">
  440. <label class="md-nav__title" for="__nav_6">
  441. <span class="md-nav__icon md-icon"></span>
  442. ⑤ RAG系统评估
  443. </label>
  444. <ul class="md-nav__list" data-md-scrollfix>
  445. <li class="md-nav__item">
  446. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.1%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0%E5%B7%A5%E5%85%B7.html" class="md-nav__link">
  447. <span class="md-ellipsis">
  448. 5.1 RAG系统评估工具
  449. </span>
  450. </a>
  451. </li>
  452. <li class="md-nav__item">
  453. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.2%20%E5%AE%9E%E7%8E%B0EnterpriseKB-RAG%E7%B3%BB%E7%BB%9F%E7%9A%84%E8%AF%84%E4%BC%B0.html" class="md-nav__link">
  454. <span class="md-ellipsis">
  455. 5.2 实现企业知识库RAG系统的评估
  456. </span>
  457. </a>
  458. </li>
  459. <li class="md-nav__item">
  460. <a href="../%E2%91%A4%20RAG%E7%B3%BB%E7%BB%9F%E8%AF%84%E4%BC%B0/5.3%20RAG%E8%AF%84%E4%BC%B0%E6%95%B0%E6%8D%AE%E9%9B%86%E7%9A%84%E6%9E%84%E9%80%A0%28%E6%89%A9%E5%B1%95%29.html" class="md-nav__link">
  461. <span class="md-ellipsis">
  462. 5.3 RAG评估数据集的构造(扩展)
  463. </span>
  464. </a>
  465. </li>
  466. </ul>
  467. </nav>
  468. </li>
  469. <li class="md-nav__item md-nav__item--nested">
  470. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_7">
  471. <label class="md-nav__link" for="__nav_7" id="__nav_7_label" tabindex="0">
  472. <span class="md-ellipsis">
  473. ⑥ 融合Mysql的RAG系统
  474. </span>
  475. <span class="md-nav__icon md-icon"></span>
  476. </label>
  477. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_7_label" aria-expanded="false">
  478. <label class="md-nav__title" for="__nav_7">
  479. <span class="md-nav__icon md-icon"></span>
  480. ⑥ 融合Mysql的RAG系统
  481. </label>
  482. <ul class="md-nav__list" data-md-scrollfix>
  483. <li class="md-nav__item">
  484. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.1%20%E8%9E%8D%E5%90%88FQA%E5%92%8C%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2.html" class="md-nav__link">
  485. <span class="md-ellipsis">
  486. 6.1 融合FQA和知识库查询
  487. </span>
  488. </a>
  489. </li>
  490. <li class="md-nav__item">
  491. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.2%20%E8%9E%8D%E5%90%88FAQ%20%E5%92%8C%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2%EF%BC%88%E4%BC%98%E5%8C%96%E7%89%88%EF%BC%89.html" class="md-nav__link">
  492. <span class="md-ellipsis">
  493. 6.2 融合FAQ 和知识库查询(优化版)
  494. </span>
  495. </a>
  496. </li>
  497. <li class="md-nav__item">
  498. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.3%20%E5%9F%BA%E4%BA%8E%20FastAPI%20%E7%9A%84%20FAQ%20%E4%B8%8E%E7%9F%A5%E8%AF%86%E5%BA%93%E6%9F%A5%E8%AF%A2%E6%8E%A5%E5%8F%A3.html" class="md-nav__link">
  499. <span class="md-ellipsis">
  500. 6.3 基于 FastAPI 的 FAQ 与知识库查询接口
  501. </span>
  502. </a>
  503. </li>
  504. <li class="md-nav__item">
  505. <a href="../%E2%91%A5%20%E8%9E%8D%E5%90%88Mysql%E7%9A%84RAG%E7%B3%BB%E7%BB%9F/6.4%20FastAPI%E5%92%8CFlask%E7%9A%84%E5%8C%BA%E5%88%AB%E5%92%8C%E8%81%94%E7%B3%BB%28%E6%89%A9%E5%B1%95%E8%B5%84%E6%96%99%29.html" class="md-nav__link">
  506. <span class="md-ellipsis">
  507. 6.4 FastAPI和Flask的区别和联系(扩展资料)
  508. </span>
  509. </a>
  510. </li>
  511. </ul>
  512. </nav>
  513. </li>
  514. <li class="md-nav__item md-nav__item--nested">
  515. <input class="md-nav__toggle md-toggle " type="checkbox" id="__nav_8">
  516. <label class="md-nav__link" for="__nav_8" id="__nav_8_label" tabindex="0">
  517. <span class="md-ellipsis">
  518. ⑦ 企业级生产环境部署
  519. </span>
  520. <span class="md-nav__icon md-icon"></span>
  521. </label>
  522. <nav class="md-nav" data-md-level="1" aria-labelledby="__nav_8_label" aria-expanded="false">
  523. <label class="md-nav__title" for="__nav_8">
  524. <span class="md-nav__icon md-icon"></span>
  525. ⑦ 企业级生产环境部署
  526. </label>
  527. <ul class="md-nav__list" data-md-scrollfix>
  528. <li class="md-nav__item">
  529. <a href="../%E2%91%A6%20%E4%BC%81%E4%B8%9A%E7%BA%A7%E7%94%9F%E4%BA%A7%E7%8E%AF%E5%A2%83%E9%83%A8%E7%BD%B2/7.1%20Docker%E7%9A%84%E5%8E%9F%E7%90%86%E5%92%8C%E5%9F%BA%E6%9C%AC%E4%BD%BF%E7%94%A8.html" class="md-nav__link">
  530. <span class="md-ellipsis">
  531. 7.1 Docker的原理和基本使用
  532. </span>
  533. </a>
  534. </li>
  535. <li class="md-nav__item">
  536. <a href="../%E2%91%A6%20%E4%BC%81%E4%B8%9A%E7%BA%A7%E7%94%9F%E4%BA%A7%E7%8E%AF%E5%A2%83%E9%83%A8%E7%BD%B2/7.2%20%E6%9C%8D%E5%8A%A1%E9%83%A8%E7%BD%B2.html" class="md-nav__link">
  537. <span class="md-ellipsis">
  538. 7.2 服务部署
  539. </span>
  540. </a>
  541. </li>
  542. </ul>
  543. </nav>
  544. </li>
  545. </ul>
  546. </nav>
  547. </div>
  548. </div>
  549. </div>
  550. <div class="md-sidebar md-sidebar--secondary" data-md-component="sidebar" data-md-type="toc">
  551. <div class="md-sidebar__scrollwrap">
  552. <div class="md-sidebar__inner">
  553. <nav class="md-nav md-nav--secondary" aria-label="目录">
  554. <label class="md-nav__title" for="__toc">
  555. <span class="md-nav__icon md-icon"></span>
  556. 目录
  557. </label>
  558. <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix>
  559. <li class="md-nav__item">
  560. <a href="#_1" class="md-nav__link">
  561. <span class="md-ellipsis">
  562. 学习目标
  563. </span>
  564. </a>
  565. </li>
  566. <li class="md-nav__item">
  567. <a href="#bm25" class="md-nav__link">
  568. <span class="md-ellipsis">
  569. 一、BM25算法概述
  570. </span>
  571. </a>
  572. <nav class="md-nav" aria-label="一、BM25算法概述">
  573. <ul class="md-nav__list">
  574. <li class="md-nav__item">
  575. <a href="#1" class="md-nav__link">
  576. <span class="md-ellipsis">
  577. 1 公式
  578. </span>
  579. </a>
  580. </li>
  581. <li class="md-nav__item">
  582. <a href="#2" class="md-nav__link">
  583. <span class="md-ellipsis">
  584. 2 简单示例
  585. </span>
  586. </a>
  587. </li>
  588. </ul>
  589. </nav>
  590. </li>
  591. <li class="md-nav__item">
  592. <a href="#_2" class="md-nav__link">
  593. <span class="md-ellipsis">
  594. 二、代码实现
  595. </span>
  596. </a>
  597. <nav class="md-nav" aria-label="二、代码实现">
  598. <ul class="md-nav__list">
  599. <li class="md-nav__item">
  600. <a href="#1_1" class="md-nav__link">
  601. <span class="md-ellipsis">
  602. 1 整体结构
  603. </span>
  604. </a>
  605. </li>
  606. <li class="md-nav__item">
  607. <a href="#2_1" class="md-nav__link">
  608. <span class="md-ellipsis">
  609. 2 具体模块
  610. </span>
  611. </a>
  612. <nav class="md-nav" aria-label="2 具体模块">
  613. <ul class="md-nav__list">
  614. <li class="md-nav__item">
  615. <a href="#21-retrievalbm25_searchpy" class="md-nav__link">
  616. <span class="md-ellipsis">
  617. 2.1 检索模块 (retrieval/bm25_search.py)
  618. </span>
  619. </a>
  620. </li>
  621. <li class="md-nav__item">
  622. <a href="#22-mainpy" class="md-nav__link">
  623. <span class="md-ellipsis">
  624. 2.2 主程序 (main.py)
  625. </span>
  626. </a>
  627. </li>
  628. <li class="md-nav__item">
  629. <a href="#23-requirementstxt" class="md-nav__link">
  630. <span class="md-ellipsis">
  631. 2.3 依赖文件 (requirements.txt)
  632. </span>
  633. </a>
  634. </li>
  635. </ul>
  636. </nav>
  637. </li>
  638. </ul>
  639. </nav>
  640. </li>
  641. <li class="md-nav__item">
  642. <a href="#_3" class="md-nav__link">
  643. <span class="md-ellipsis">
  644. 三、示例运行结果
  645. </span>
  646. </a>
  647. </li>
  648. <li class="md-nav__item">
  649. <a href="#_4" class="md-nav__link">
  650. <span class="md-ellipsis">
  651. 本章小结
  652. </span>
  653. </a>
  654. </li>
  655. </ul>
  656. </nav>
  657. </div>
  658. </div>
  659. </div>
  660. <div class="md-content" data-md-component="content">
  661. <article class="md-content__inner md-typeset">
  662. <h1>3.4 BM25算法简介与应用</h1>
  663. <h2 id="_1">学习目标<a class="headerlink" href="#_1" title="Permanent link">¶</a></h2>
  664. <ol>
  665. <li>理解BM25算法的基本原理及其在信息检索中的作用。</li>
  666. <li>掌握如何使用BM25进行文本匹配。</li>
  667. <li>学会将BM25算法集成到工程化代码中。</li>
  668. </ol>
  669. <hr>
  670. <h2 id="bm25">一、BM25算法概述<a class="headerlink" href="#bm25" title="Permanent link">¶</a></h2>
  671. <h3 id="1">1 公式<a class="headerlink" href="#1" title="Permanent link">¶</a></h3>
  672. <p>BM25(Best Matching 25)是一种信息检索领域的排名算法,用于计算查询(Query)与文档(Document)之间的相关性得分。它改进了传统的TF-IDF算法,引入文档长度归一化和词频饱和机制,使检索结果更准确。</p>
  673. <div align="center"><img src="../img/3-2-1.png" style="zoom:90%"><img></div>
  674. <p>其中:</p>
  675. <blockquote>
  676. <ul>
  677. <li>fi 是词项 qi在文档 D中的出现次数(TF),</li>
  678. <li>N是文档总数,</li>
  679. <li>ni是包含词项 qi的文档数,</li>
  680. <li>∣D∣是文档 D的长度,</li>
  681. <li>avgdl 是所有文档的平均长度,</li>
  682. <li>k1 和 b是调整参数,通常设置为 k1=1.2和 b=0.75 。</li>
  683. </ul>
  684. </blockquote>
  685. <h3 id="2">2 简单示例<a class="headerlink" href="#2" title="Permanent link">¶</a></h3>
  686. <p>假设我们有以下文档集合:</p>
  687. <ul>
  688. <li>
  689. <p>文档1:"我喜欢编程"</p>
  690. </li>
  691. <li>
  692. <p>文档2:"编程很有趣"</p>
  693. </li>
  694. <li>查询:"他喜欢编程"</li>
  695. </ul>
  696. <p><strong>步骤</strong>:</p>
  697. <ol>
  698. <li>分词:将文档和查询拆分为词。</li>
  699. <li>文档1:["我", "喜欢", "编程"]</li>
  700. <li>文档2:["编程", "很", "有趣"]</li>
  701. <li>查询:["他", "喜欢", "编程"]</li>
  702. <li>计算BM25得分:使用<code>rank_bm25</code>库计算每个文档与查询的相关性。</li>
  703. </ol>
  704. <hr>
  705. <h2 id="_2">二、代码实现<a class="headerlink" href="#_2" title="Permanent link">¶</a></h2>
  706. <h3 id="1_1">1 整体结构<a class="headerlink" href="#1_1" title="Permanent link">¶</a></h3>
  707. <div class="highlight"><pre><span></span><code>bm25_lesson/
  708. ├── retrieval/
  709. │ └── bm25_search.py # BM25检索模块
  710. ├── main.py # 主程序入口
  711. └── requirements.txt # 依赖文件
  712. </code></pre></div>
  713. <h3 id="2_1">2 具体模块<a class="headerlink" href="#2_1" title="Permanent link">¶</a></h3>
  714. <h4 id="21-retrievalbm25_searchpy">2.1 检索模块 (<code>retrieval/bm25_search.py</code>)<a class="headerlink" href="#21-retrievalbm25_searchpy" title="Permanent link">¶</a></h4>
  715. <div class="highlight"><pre><span></span><code><span class="kn">import</span><span class="w"> </span><span class="nn">jieba</span>
  716. <span class="kn">from</span><span class="w"> </span><span class="nn">rank_bm25</span><span class="w"> </span><span class="kn">import</span> <span class="n">BM25L</span>
  717. <span class="kn">import</span><span class="w"> </span><span class="nn">logging</span>
  718. <span class="c1"># 配置日志</span>
  719. <span class="n">logging</span><span class="o">.</span><span class="n">basicConfig</span><span class="p">(</span><span class="n">level</span><span class="o">=</span><span class="n">logging</span><span class="o">.</span><span class="n">INFO</span><span class="p">,</span> <span class="nb">format</span><span class="o">=</span><span class="s1">'</span><span class="si">%(asctime)s</span><span class="s1"> - </span><span class="si">%(levelname)s</span><span class="s1"> - </span><span class="si">%(message)s</span><span class="s1">'</span><span class="p">)</span>
  720. <span class="n">logger</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="vm">__name__</span><span class="p">)</span>
  721. <span class="k">class</span><span class="w"> </span><span class="nc">BM25Search</span><span class="p">:</span>
  722. <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">documents</span><span class="p">):</span>
  723. <span class="c1"># 初始化文档集合</span>
  724. <span class="bp">self</span><span class="o">.</span><span class="n">documents</span> <span class="o">=</span> <span class="n">documents</span>
  725. <span class="c1"># 分词后的文档</span>
  726. <span class="bp">self</span><span class="o">.</span><span class="n">tokenized_docs</span> <span class="o">=</span> <span class="p">[</span><span class="n">jieba</span><span class="o">.</span><span class="n">lcut</span><span class="p">(</span><span class="n">doc</span><span class="p">)</span> <span class="k">for</span> <span class="n">doc</span> <span class="ow">in</span> <span class="n">documents</span><span class="p">]</span>
  727. <span class="c1"># 初始化BM25模型</span>
  728. <span class="bp">self</span><span class="o">.</span><span class="n">bm25</span> <span class="o">=</span> <span class="n">BM25L</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">tokenized_docs</span><span class="p">)</span>
  729. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"BM25模型初始化完成"</span><span class="p">)</span>
  730. <span class="k">def</span><span class="w"> </span><span class="nf">search</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">query</span><span class="p">):</span>
  731. <span class="c1"># 分词查询</span>
  732. <span class="n">tokenized_query</span> <span class="o">=</span> <span class="n">jieba</span><span class="o">.</span><span class="n">lcut</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  733. <span class="k">try</span><span class="p">:</span>
  734. <span class="c1"># 计算每个文档的BM25得分</span>
  735. <span class="n">scores</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">bm25</span><span class="o">.</span><span class="n">get_scores</span><span class="p">(</span><span class="n">tokenized_query</span><span class="p">)</span>
  736. <span class="nb">print</span><span class="p">(</span><span class="sa">f</span><span class="s1">'scores--》</span><span class="si">{</span><span class="n">scores</span><span class="si">}</span><span class="s1">'</span><span class="p">)</span>
  737. <span class="c1"># 获取最高得分的文档索引</span>
  738. <span class="n">best_idx</span> <span class="o">=</span> <span class="n">scores</span><span class="o">.</span><span class="n">argmax</span><span class="p">()</span>
  739. <span class="n">best_score</span> <span class="o">=</span> <span class="n">scores</span><span class="p">[</span><span class="n">best_idx</span><span class="p">]</span>
  740. <span class="n">best_doc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">documents</span><span class="p">[</span><span class="n">best_idx</span><span class="p">]</span>
  741. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"查询: </span><span class="si">{</span><span class="n">query</span><span class="si">}</span><span class="s2">, 最佳匹配: </span><span class="si">{</span><span class="n">best_doc</span><span class="si">}</span><span class="s2">, 得分: </span><span class="si">{</span><span class="n">best_score</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  742. <span class="k">return</span> <span class="n">best_doc</span><span class="p">,</span> <span class="n">best_score</span>
  743. <span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
  744. <span class="n">logger</span><span class="o">.</span><span class="n">error</span><span class="p">(</span><span class="sa">f</span><span class="s2">"检索失败: </span><span class="si">{</span><span class="n">e</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  745. <span class="k">return</span> <span class="kc">None</span><span class="p">,</span> <span class="mf">0.0</span>
  746. </code></pre></div>
  747. <h4 id="22-mainpy">2.2 主程序 (<code>main.py</code>)<a class="headerlink" href="#22-mainpy" title="Permanent link">¶</a></h4>
  748. <div class="highlight"><pre><span></span><code><span class="kn">from</span><span class="w"> </span><span class="nn">retrieval.bm25_search</span><span class="w"> </span><span class="kn">import</span> <span class="n">BM25Search</span>
  749. <span class="kn">import</span><span class="w"> </span><span class="nn">logging</span>
  750. <span class="c1"># 配置日志</span>
  751. <span class="n">logging</span><span class="o">.</span><span class="n">basicConfig</span><span class="p">(</span><span class="n">level</span><span class="o">=</span><span class="n">logging</span><span class="o">.</span><span class="n">INFO</span><span class="p">,</span> <span class="nb">format</span><span class="o">=</span><span class="s1">'</span><span class="si">%(asctime)s</span><span class="s1"> - </span><span class="si">%(levelname)s</span><span class="s1"> - </span><span class="si">%(message)s</span><span class="s1">'</span><span class="p">)</span>
  752. <span class="n">logger</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="vm">__name__</span><span class="p">)</span>
  753. <span class="k">def</span><span class="w"> </span><span class="nf">main</span><span class="p">():</span>
  754. <span class="c1"># 示例文档集合</span>
  755. <span class="n">documents</span> <span class="o">=</span> <span class="p">[</span><span class="s2">"我喜欢编程"</span><span class="p">,</span> <span class="s2">"编程很有趣"</span><span class="p">]</span>
  756. <span class="c1"># 初始化BM25检索器</span>
  757. <span class="n">bm25_search</span> <span class="o">=</span> <span class="n">BM25Search</span><span class="p">(</span><span class="n">documents</span><span class="p">)</span>
  758. <span class="c1"># 示例查询</span>
  759. <span class="n">query</span> <span class="o">=</span> <span class="s2">"他喜欢编程"</span>
  760. <span class="c1"># 执行检索</span>
  761. <span class="n">result</span><span class="p">,</span> <span class="n">score</span> <span class="o">=</span> <span class="n">bm25_search</span><span class="o">.</span><span class="n">search</span><span class="p">(</span><span class="n">query</span><span class="p">)</span>
  762. <span class="k">if</span> <span class="n">result</span><span class="p">:</span>
  763. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="sa">f</span><span class="s2">"查询结果: </span><span class="si">{</span><span class="n">result</span><span class="si">}</span><span class="s2">, 得分: </span><span class="si">{</span><span class="n">score</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
  764. <span class="k">else</span><span class="p">:</span>
  765. <span class="n">logger</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="s2">"未找到匹配结果"</span><span class="p">)</span>
  766. <span class="k">if</span> <span class="vm">__name__</span> <span class="o">==</span> <span class="s2">"__main__"</span><span class="p">:</span>
  767. <span class="n">main</span><span class="p">()</span>
  768. </code></pre></div>
  769. <h4 id="23-requirementstxt">2.3 依赖文件 (<code>requirements.txt</code>)<a class="headerlink" href="#23-requirementstxt" title="Permanent link">¶</a></h4>
  770. <div class="highlight"><pre><span></span><code>jieba
  771. rank_bm25
  772. </code></pre></div>
  773. <hr>
  774. <h2 id="_3">三、示例运行结果<a class="headerlink" href="#_3" title="Permanent link">¶</a></h2>
  775. <p>运行<code>main.py</code>,输出如下:
  776. </p><div class="highlight"><pre><span></span><code>2025-04-02 19:01:27,463 - INFO - BM25模型初始化完成
  777. 2025-04-02 19:01:27,464 - INFO - 查询: 他喜欢编程, 最佳匹配: 我喜欢编程, 得分: 1.094
  778. 2025-04-02 19:01:27,464 - INFO - 查询结果: 我喜欢编程, 得分: 1.094
  779. </code></pre></div>
  780. <p>分析:</p>
  781. <ul>
  782. <li>"编程很有趣"得分高于"我喜欢编程",因为前者词频更高且文档较短。</li>
  783. <li>BM25通过长度归一化避免了长文档的过度优势。</li>
  784. </ul>
  785. <hr>
  786. <h2 id="_4">本章小结<a class="headerlink" href="#_4" title="Permanent link">¶</a></h2>
  787. <p>本节主要介绍了BM25算法的原理和应用:</p>
  788. <ul>
  789. <li>
  790. <p><strong>原理</strong>:结合TF、IDF、长度归一化和词频饱和。</p>
  791. </li>
  792. <li>
  793. <p><strong>应用</strong>:通过<code>rank_bm25</code>库实现文本检索。</p>
  794. </li>
  795. </ul>
  796. <hr>
  797. </article>
  798. </div>
  799. <script>var target=document.getElementById(location.hash.slice(1));target&&target.name&&(target.checked=target.name.startsWith("__tabbed_"))</script>
  800. </div>
  801. </main>
  802. <footer class="md-footer">
  803. <div class="md-footer-meta md-typeset">
  804. <div class="md-footer-meta__inner md-grid">
  805. <div class="md-copyright">
  806. <div class="md-copyright__highlight">
  807. © EnterpriseKB-RAG 2026. Internal knowledge base demo site.
  808. </div>
  809. Made with
  810. <a href="https://squidfunk.github.io/mkdocs-material/" target="_blank" rel="noopener">
  811. Material for MkDocs
  812. </a>
  813. </div>
  814. </div>
  815. </div>
  816. </footer>
  817. </div>
  818. <div class="md-dialog" data-md-component="dialog">
  819. <div class="md-dialog__inner md-typeset"></div>
  820. </div>
  821. <script id="__config" type="application/json">{"base": "..", "features": [], "search": "../assets/javascripts/workers/search.973d3a69.min.js", "tags": null, "translations": {"clipboard.copied": "\u5df2\u590d\u5236", "clipboard.copy": "\u590d\u5236", "search.result.more.one": "\u5728\u8be5\u9875\u4e0a\u8fd8\u6709 1 \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.more.other": "\u5728\u8be5\u9875\u4e0a\u8fd8\u6709 # \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.none": "\u6ca1\u6709\u627e\u5230\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.one": "\u627e\u5230 1 \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.other": "# \u4e2a\u7b26\u5408\u6761\u4ef6\u7684\u7ed3\u679c", "search.result.placeholder": "\u952e\u5165\u4ee5\u5f00\u59cb\u641c\u7d22", "search.result.term.missing": "\u7f3a\u5c11", "select.version": "\u9009\u62e9\u5f53\u524d\u7248\u672c"}, "version": null}</script>
  822. <script src="../assets/javascripts/bundle.f55a23d4.min.js"></script>
  823. <script src="../javascripts/extra.js"></script>
  824. <script src="../javascripts/mathjax.js"></script>
  825. <!-- <script src="https://polyfill.io/v3/polyfill.min.js?features=es6"></script>-->
  826. <script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
  827. </body>
  828. </html>