Fix loss unit test

Signed-off-by: cwj <[email protected]>
FederatedAI · Jun 14, 2023 · 081779c · 081779c
1 parent 1f72e64
commit 081779c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ratedml/ensemble/basic_algorithms/decision_tree/tree_core/loss/test/cross_entropy_test.py b/...ratedml/ensemble/basic_algorithms/decision_tree/tree_core/loss/test/cross_entropy_test.py
@@ -96,7 +96,7 @@ def test_compute_hess(self):
             pred = np.asarray([random.random() for j in range(5)], dtype='float64')
             label = random.randint(0, 4)
             softmaxloss_hess = self.softmax_loss.compute_hess(label, pred)
-            hess = pred * (1 - pred)
+            hess = 2 * pred * (1 - pred)
             self.assertTrue(np.fabs(hess - softmaxloss_hess).all() < consts.FLOAT_ZERO)
 
     def test_compute_loss(self):