Internal change

PiperOrigin-RevId: 523429555
This commit is contained in:
Sebastian Schmidt 2023-04-11 09:54:25 -07:00 committed by Copybara-Service
parent a448790300
commit b66b0e0c72
2 changed files with 5 additions and 5 deletions

View File

@ -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"});

View File

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