From 2687f2afec9437bbd89eb55a0da9e135eb53546b Mon Sep 17 00:00:00 2001 From: lijianshe02 <48898730+lijianshe02@users.noreply.github.com> Date: Thu, 26 Dec 2019 19:04:43 +0800 Subject: [PATCH] fix input shape of test_log_loss_op (#21953) --- .../paddle/fluid/tests/unittests/test_log_loss_op.py | 12 ++++++++---- .../unittests/white_list/check_shape_white_list.py | 2 +- 2 files changed, 9 insertions(+), 5 deletions(-) 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 784f4f648d..27da4d1816 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 6a8df02164..a7426bb516 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' ] -- GitLab