Skip to content

Commit 73ea396

Browse files
talumbaucopybara-github
authored andcommitted
New goldens for text embedder, classifier tests
PiperOrigin-RevId: 666887484
1 parent 46b800a commit 73ea396

File tree

4 files changed

+8
-14
lines changed

4 files changed

+8
-14
lines changed

mediapipe/tasks/c/text/text_classifier/text_classifier_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ TEST(TextClassifierTest, SmokeTest) {
6565
EXPECT_EQ(result.classifications[0].categories_count, 2);
6666
EXPECT_EQ(std::string{result.classifications[0].categories[0].category_name},
6767
"positive");
68-
EXPECT_NEAR(result.classifications[0].categories[0].score, 0.999371,
68+
EXPECT_NEAR(result.classifications[0].categories[0].score, 0.999465,
6969
kPrecision);
7070

7171
text_classifier_close_result(&result);

mediapipe/tasks/c/text/text_embedder/text_embedder_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ TEST(TextEmbedderTest, SucceedsWithCosineSimilarity) {
9191
double similarity;
9292
text_embedder_cosine_similarity(&result0.embeddings[0],
9393
&result1.embeddings[0], &similarity, nullptr);
94-
double expected_similarity = 0.98077;
94+
double expected_similarity = 0.97565;
9595
EXPECT_LE(abs(similarity - expected_similarity), kPrecision);
9696

9797
text_embedder_close_result(&result0);

mediapipe/tasks/cc/text/text_classifier/text_classifier_test.cc

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -151,15 +151,11 @@ TEST_F(TextClassifierTest, TextClassifierWithBert) {
151151
/*head_name=*/"probability"});
152152
#else
153153
negative_expected.classifications.emplace_back(Classifications{
154-
/*categories=*/{
155-
{/*index=*/0, /*score=*/0.956316, /*category_name=*/"negative"},
156-
{/*index=*/1, /*score=*/0.043683, /*category_name=*/"positive"}},
154+
/*categories=*/{{0, 0.963325, "negative"}, {1, 0.036674, "positive"}},
157155
/*head_index=*/0,
158156
/*head_name=*/"probability"});
159157
positive_expected.classifications.emplace_back(Classifications{
160-
/*categories=*/{
161-
{/*index=*/1, /*score=*/0.999945, /*category_name=*/"positive"},
162-
{/*index=*/0, /*score=*/0.000056, /*category_name=*/"negative"}},
158+
/*categories=*/{{1, 0.9999413, "positive"}, {0, 0.000058, "negative"}},
163159
/*head_index=*/0,
164160
/*head_name=*/"probability"});
165161
#endif // _WIN32
@@ -255,10 +251,8 @@ TEST_F(TextClassifierTest, BertLongPositive) {
255251
categories.push_back(
256252
{/*index=*/0, /*score=*/0.023313, /*category_name=*/"negative"});
257253
#else
258-
categories.push_back(
259-
{/*index=*/1, /*score=*/0.985889, /*category_name=*/"positive"});
260-
categories.push_back(
261-
{/*index=*/0, /*score=*/0.014112, /*category_name=*/"negative"});
254+
categories.push_back({1, 0.981590, "positive"});
255+
categories.push_back({0, 0.018409, "negative"});
262256
#endif // _WIN32
263257

264258
expected.classifications.emplace_back(

mediapipe/tasks/cc/text/text_embedder/text_embedder_test.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ TEST_F(EmbedderTest, SucceedsWithMobileBert) {
8282
#elif defined(__FMA__)
8383
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 21.3605f, kEpsilon);
8484
#else
85-
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 19.9016f, kEpsilon);
85+
ASSERT_NEAR(result0.embeddings[0].float_embedding[0], 21.1785f, kEpsilon);
8686
#endif // _WIN32
8787

8888
MP_ASSERT_OK_AND_ASSIGN(
@@ -92,7 +92,7 @@ TEST_F(EmbedderTest, SucceedsWithMobileBert) {
9292
#ifdef __FMA__
9393
ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 21.254150f, kEpsilon);
9494
#else
95-
ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 22.626251f, kEpsilon);
95+
ASSERT_NEAR(result1.embeddings[0].float_embedding[0], 20.322639f, kEpsilon);
9696
#endif
9797

9898
// Check cosine similarity.

0 commit comments

Comments
 (0)