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 #else
negative_expected.classifications.emplace_back(Classifications{ negative_expected.classifications.emplace_back(Classifications{
/*categories=*/{ /*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"}}, {/*index=*/1, /*score=*/0.043683, /*category_name=*/"positive"}},
/*head_index=*/0, /*head_index=*/0,
/*head_name=*/"probability"}); /*head_name=*/"probability"});

View File

@ -46,12 +46,12 @@ _BERT_NEGATIVE_RESULTS = TextClassifierResult(
categories=[ categories=[
_Category( _Category(
index=0, index=0,
score=0.999479, score=0.9995,
display_name='', display_name='',
category_name='negative'), category_name='negative'),
_Category( _Category(
index=1, index=1,
score=0.00052154, score=0.0005,
display_name='', display_name='',
category_name='positive') category_name='positive')
], ],
@ -65,12 +65,12 @@ _BERT_POSITIVE_RESULTS = TextClassifierResult(
categories=[ categories=[
_Category( _Category(
index=1, index=1,
score=0.999466, score=0.9995,
display_name='', display_name='',
category_name='positive'), category_name='positive'),
_Category( _Category(
index=0, index=0,
score=0.000533596, score=0.0005,
display_name='', display_name='',
category_name='negative') category_name='negative')
], ],