From b66b0e0c723cf6321ef1e0f568f185a383c5b2a0 Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Tue, 11 Apr 2023 09:54:25 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 523429555 --- .../tasks/cc/text/text_classifier/text_classifier_test.cc | 2 +- mediapipe/tasks/python/test/text/text_classifier_test.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/mediapipe/tasks/cc/text/text_classifier/text_classifier_test.cc b/mediapipe/tasks/cc/text/text_classifier/text_classifier_test.cc index 70094dd7..98ca9e90 100644 --- a/mediapipe/tasks/cc/text/text_classifier/text_classifier_test.cc +++ b/mediapipe/tasks/cc/text/text_classifier/text_classifier_test.cc @@ -155,7 +155,7 @@ TEST_F(TextClassifierTest, TextClassifierWithBert) { #else negative_expected.classifications.emplace_back(Classifications{ /*categories=*/{ - {/*index=*/0, /*score=*/0.956317, /*category_name=*/"negative"}, + {/*index=*/0, /*score=*/0.956316, /*category_name=*/"negative"}, {/*index=*/1, /*score=*/0.043683, /*category_name=*/"positive"}}, /*head_index=*/0, /*head_name=*/"probability"}); diff --git a/mediapipe/tasks/python/test/text/text_classifier_test.py b/mediapipe/tasks/python/test/text/text_classifier_test.py index 8df7dce8..6fea4c6f 100644 --- a/mediapipe/tasks/python/test/text/text_classifier_test.py +++ b/mediapipe/tasks/python/test/text/text_classifier_test.py @@ -46,12 +46,12 @@ _BERT_NEGATIVE_RESULTS = TextClassifierResult( categories=[ _Category( index=0, - score=0.999479, + score=0.9995, display_name='', category_name='negative'), _Category( index=1, - score=0.00052154, + score=0.0005, display_name='', category_name='positive') ], @@ -65,12 +65,12 @@ _BERT_POSITIVE_RESULTS = TextClassifierResult( categories=[ _Category( index=1, - score=0.999466, + score=0.9995, display_name='', category_name='positive'), _Category( index=0, - score=0.000533596, + score=0.0005, display_name='', category_name='negative') ],