added reranking comparison image

This commit is contained in:
nird
2024-08-30 13:57:03 +03:00
parent 0f95f187a0
commit 54f6339472
2 changed files with 92 additions and 0 deletions

View File

@@ -58,6 +58,16 @@
"</div>"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"<div style=\"text-align: center;\">\n",
"\n",
"<img src=\"../images/reranking_comparison.svg\" alt=\"rerank llm\" style=\"width:100%; height:auto;\">\n",
"</div>"
]
},
{
"cell_type": "markdown",
"metadata": {},

View File

@@ -0,0 +1,82 @@
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 800 650">
<defs>
<filter id="shadow" x="-20%" y="-20%" width="140%" height="140%">
<feDropShadow dx="2" dy="2" stdDeviation="2" flood-color="#000000" flood-opacity="0.3"/>
</filter>
</defs>
<!-- Background -->
<rect x="0" y="0" width="800" height="650" fill="#f0f9ff" rx="20" ry="20" />
<!-- Title -->
<text x="400" y="40" text-anchor="middle" font-size="24" font-weight="bold" fill="#333">RAG Retrieval Comparison</text>
<!-- Query -->
<text x="400" y="70" text-anchor="middle" font-size="16" fill="#555">Query: What is the capital of France?</text>
<!-- Document Collection -->
<rect x="20" y="100" width="240" height="480" fill="#ffffff" rx="10" ry="10" filter="url(#shadow)" />
<text x="140" y="130" text-anchor="middle" font-size="18" font-weight="bold" fill="#333">Document Collection</text>
<!-- Baseline Retrieval -->
<rect x="280" y="100" width="240" height="480" fill="#ffffff" rx="10" ry="10" filter="url(#shadow)" />
<text x="400" y="130" text-anchor="middle" font-size="18" font-weight="bold" fill="#333">Baseline Retrieval</text>
<!-- Advanced Retrieval -->
<rect x="540" y="100" width="240" height="480" fill="#ffffff" rx="10" ry="10" filter="url(#shadow)" />
<text x="660" y="130" text-anchor="middle" font-size="18" font-weight="bold" fill="#333">Advanced Retrieval</text>
<!-- Documents -->
<g id="doc1">
<rect x="30" y="150" width="220" height="80" fill="#e6f3ff" rx="5" ry="5" />
<text x="40" y="170" font-size="12" fill="#333">The capital of France is great.</text>
</g>
<g id="doc2">
<rect x="30" y="240" width="220" height="80" fill="#e6f3ff" rx="5" ry="5" />
<text x="40" y="260" font-size="12" fill="#333">The capital of France is huge.</text>
</g>
<g id="doc3">
<rect x="30" y="330" width="220" height="80" fill="#e6f3ff" rx="5" ry="5" />
<text x="40" y="350" font-size="12" fill="#333">The capital of France is beautiful.</text>
</g>
<g id="doc4">
<rect x="30" y="420" width="220" height="80" fill="#e6f3ff" rx="5" ry="5" />
<text x="40" y="440" font-size="12" fill="#333">
<tspan x="40" dy="0">Have you ever visited Paris? It is</tspan>
<tspan x="40" dy="14">a beautiful city where you can</tspan>
<tspan x="40" dy="14">eat delicious food and see the</tspan>
<tspan x="40" dy="14">Eiffel Tower...</tspan>
</text>
</g>
<g id="doc5">
<rect x="30" y="510" width="220" height="80" fill="#e6f3ff" rx="5" ry="5" />
<text x="40" y="530" font-size="12" fill="#333">
<tspan x="40" dy="0">I really enjoyed my trip to Paris,</tspan>
<tspan x="40" dy="14">France. The city is beautiful and</tspan>
<tspan x="40" dy="14">the food is delicious. I would love</tspan>
<tspan x="40" dy="14">to visit again...</tspan>
</text>
</g>
<!-- Baseline Results -->
<use href="#doc1" x="250" />
<use href="#doc3" x="250" />
<rect x="280" y="150" width="240" height="80" fill="none" stroke="#4e79a7" stroke-width="3" rx="5" ry="5" />
<rect x="280" y="330" width="240" height="80" fill="none" stroke="#4e79a7" stroke-width="3" rx="5" ry="5" />
<!-- Advanced Results -->
<use href="#doc4" x="510" />
<use href="#doc5" x="510" />
<rect x="540" y="420" width="240" height="80" fill="none" stroke="#e15759" stroke-width="3" rx="5" ry="5" />
<rect x="540" y="510" width="240" height="80" fill="none" stroke="#e15759" stroke-width="3" rx="5" ry="5" />
<!-- Legend (Moved to the bottom) -->
<rect x="250" y="600" width="20" height="20" fill="none" stroke="#4e79a7" stroke-width="3" />
<text x="280" y="615" font-size="14" fill="#333">Baseline Retrieval</text>
<rect x="470" y="600" width="20" height="20" fill="none" stroke="#e15759" stroke-width="3" />
<text x="500" y="615" font-size="14" fill="#333">Advanced Retrieval</text>
</svg>

After

Width:  |  Height:  |  Size: 3.8 KiB