diff --git a/python/paddle/fluid/tests/unittests/test_log_loss_op.py b/python/paddle/fluid/tests/unittests/test_log_loss_op.py index 784f4f648d52bdf4f2357f4454d790a8d53288f3..27da4d18169fed51d4860fe98451e65e2092c43b 100644 --- a/python/paddle/fluid/tests/unittests/test_log_loss_op.py +++ b/python/paddle/fluid/tests/unittests/test_log_loss_op.py @@ -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, diff --git a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py index 6a8df02164213f884c45da96a7cb5dad2157e8af..a7426bb516fcb829e653564c6fbfb44fd008288e 100644 --- a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py +++ b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py @@ -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' ]