No public description

PiperOrigin-RevId: 647279361
PAIR-code · Jun 27, 2024 · 130133a · 130133a
1 parent b3b77be
commit 130133a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/src/llm_comparator/comparison.py b/python/src/llm_comparator/comparison.py
@@ -90,6 +90,7 @@ def run(
               'score': judgement['score'],
               'individual_rater_scores': judgement['individual_rater_scores'],
               'rationale_list': similarity,
+              'custom_fields': {},
           }
           for input, judgement, similarity in per_example_generator
       ],

diff --git a/python/src/llm_comparator/rationale_cluster_generator.py b/python/src/llm_comparator/rationale_cluster_generator.py
@@ -225,7 +225,7 @@ def _store_similarities_to_rationale_bullets(
   def run(
       self,
       rationale_bullets_for_examples: Sequence[Sequence[str]],
-      num_clusters: int = 10,
+      num_clusters: int = 8,
   ) -> tuple[
       Sequence[_RationaleCluster],
       Sequence[Sequence[_RationaleBulletWithClusterSimilarity]],