Update

Matheus-Monteiro · Dec 20, 2020 · 9ed0420 · 9ed0420
1 parent fad8873
commit 9ed0420
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/deepchem/models/tests/test_losses.py b/deepchem/models/tests/test_losses.py
@@ -186,6 +186,12 @@ def test_sparse_softmax_cross_entropy_tf(self):
     expected = [-np.log(softmax[0, 1]), -np.log(softmax[1, 0])]
     assert np.allclose(expected, result)
 
+    labels = tf.constant([[1], [0]])
+    result = loss._compute_tf_loss(outputs, labels).numpy()
+    softmax = np.exp(y) / np.expand_dims(np.sum(np.exp(y), axis=1), 1)
+    expected = [-np.log(softmax[0, 1]), -np.log(softmax[1, 0])]
+    assert np.allclose(expected, result)
+
   @unittest.skipIf(not has_pytorch, 'PyTorch is not installed')
   def test_sparse_softmax_cross_entropy_pytorch(self):
     """Test SparseSoftmaxCrossEntropy."""