fix input shape of test_log_loss_op (#21953)

release/1.7
lijianshe02 5 years ago committed by Tao Luo
parent 26a5d10e75
commit 2687f2afec

@ -19,15 +19,19 @@ import numpy as np
from op_test import OpTest
def sigmoid_array(x):
return 1 / (1 + np.exp(-x))
class TestLogLossOp(OpTest):
def setUp(self):
self.op_type = 'log_loss'
samples_num = 32
samples_num = 100
predicted = np.random.uniform(0.1, 1.0,
(samples_num, 1)).astype("float32")
x = np.random.random((samples_num, 1)).astype("float32")
predicted = sigmoid_array(x)
labels = np.random.randint(0, 2, (samples_num, 1)).astype("float32")
epsilon = 1e-4
epsilon = 1e-7
self.inputs = {
'Predicted': predicted,
'Labels': labels,

@ -18,7 +18,7 @@ NOT_CHECK_OP_LIST = [
]
NEED_TO_FIX_OP_LIST = [
'sequence_scatter', 'log_loss', 'sequence_topk_avg_pooling', 'matmul',
'sequence_scatter', 'sequence_topk_avg_pooling', 'matmul',
'add_position_encoding', 'margin_rank_loss', 'elementwise_pow',
'fused_elemwise_activation', 'mul'
]

Loading…
Cancel
Save