changed error threshold (fixes LxMLS#90)

clala1 · Jun 11, 2018 · e9915a2 · e9915a2
1 parent a1bbd39
commit e9915a2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/test_non_linear_classifiers_numpy.py b/tests/test_non_linear_classifiers_numpy.py
@@ -12,8 +12,7 @@
 from lxmls.deep_learning.numpy_models.mlp import NumpyMLP
 from lxmls.deep_learning.mlp import get_mlp_parameter_handlers, get_mlp_loss_range
 
-tolerance = 2 #TODO #FIXME Implementations give random results in a +-1 margin.
-              # Check the source of this randomness
+tolerance = 1e-2
 
 @pytest.fixture(scope='module')
 def corpus():

diff --git a/tests/test_non_linear_classifiers_pytorch.py b/tests/test_non_linear_classifiers_pytorch.py
@@ -13,8 +13,7 @@
 from lxmls.deep_learning.utils import Model, glorot_weight_init
 from lxmls.deep_learning.pytorch_models.mlp import PytorchMLP
 
-tolerance = 2 #TODO #FIXME: Pytorch gives random results in a +-2 margin.
-              # Check the source of this randomness
+tolerance = 1e-2
 
 
 @pytest.fixture(scope='module')