Internal change
PiperOrigin-RevId: 523429555
This commit is contained in:
parent
a448790300
commit
b66b0e0c72
|
@ -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"});
|
||||
|
|
|
@ -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')
|
||||
],
|
||||
|
|
Loading…
Reference in New Issue
Block a user