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') ],