From 3f7752561b9fc8384daed614a37bac9774622b53 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Fri, 28 Jul 2023 12:03:14 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 551914786 --- .../mediapipe/tasks/text/textembedder/TextEmbedderTest.java | 2 +- mediapipe/tasks/python/test/text/text_embedder_test.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mediapipe/tasks/javatests/com/google/mediapipe/tasks/text/textembedder/TextEmbedderTest.java b/mediapipe/tasks/javatests/com/google/mediapipe/tasks/text/textembedder/TextEmbedderTest.java index ed7573b2..20084ee7 100644 --- a/mediapipe/tasks/javatests/com/google/mediapipe/tasks/text/textembedder/TextEmbedderTest.java +++ b/mediapipe/tasks/javatests/com/google/mediapipe/tasks/text/textembedder/TextEmbedderTest.java @@ -140,7 +140,7 @@ public class TextEmbedderTest { TextEmbedder.cosineSimilarity( result0.embeddingResult().embeddings().get(0), result1.embeddingResult().embeddings().get(0)); - assertThat(similarity).isWithin(DOUBLE_DIFF_TOLERANCE).of(0.3477488707202946); + assertThat(similarity).isWithin(DOUBLE_DIFF_TOLERANCE).of(0.3565317439544432); } @Test diff --git a/mediapipe/tasks/python/test/text/text_embedder_test.py b/mediapipe/tasks/python/test/text/text_embedder_test.py index 27726b70..9688ee91 100644 --- a/mediapipe/tasks/python/test/text/text_embedder_test.py +++ b/mediapipe/tasks/python/test/text/text_embedder_test.py @@ -37,7 +37,7 @@ _TEST_DATA_DIR = 'mediapipe/tasks/testdata/text' # Tolerance for embedding vector coordinate values. _EPSILON = 1e-4 # Tolerance for cosine similarity evaluation. -_SIMILARITY_TOLERANCE = 1e-6 +_SIMILARITY_TOLERANCE = 1e-3 class ModelFileType(enum.Enum): @@ -287,7 +287,7 @@ class TextEmbedderTest(parameterized.TestCase): @parameterized.parameters( # TODO: The similarity should likely be lower - (_BERT_MODEL_FILE, 0.980880), + (_BERT_MODEL_FILE, 0.98077), (_USE_MODEL_FILE, 0.780334), ) def test_embed_with_different_themes(self, model_file, expected_similarity):