add actual values to chart

zenml-io · Aug 8, 2024 · 28b9a9a · 28b9a9a
1 parent 548699b
commit 28b9a9a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llm-complete-guide/steps/finetune_embeddings.py b/llm-complete-guide/steps/finetune_embeddings.py
@@ -372,13 +372,17 @@ def visualize_results(
         label="Finetuned Model",
         color="red",
     )
+    for i, v in enumerate(finetuned_values):
+        ax.text(v + 1, i - height / 2, f"{v:.1f}", va="center")
     ax.barh(
         [i + height / 2 for i in y],
         base_values,
         height,
         label="Base Model",
         color="blue",
     )
+    for i, v in enumerate(base_values):
+        ax.text(v + 1, i + height / 2, f"{v:.1f}", va="center")
 
     ax.set_xlabel("Scores (%)")
     ax.set_title("Evaluation Results")