mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-06-02 14:45:15 +02:00
updated mapping test to pass gradient checks
This commit is contained in:
parent
42627f4030
commit
ef14c7a7eb
1 changed files with 1 additions and 1 deletions
|
|
@ -46,7 +46,7 @@ class MappingTests(unittest.TestCase):
|
|||
|
||||
def test_mlpextmapping(self):
|
||||
for activation in ['tanh', 'relu', 'sigmoid']:
|
||||
mapping = GPy.mappings.MLPext(input_dim=3, hidden_dims=[5,5,5], output_dim=2, activation=activation)
|
||||
mapping = GPy.mappings.MLPext(input_dim=3, hidden_dims=[5,5], output_dim=2, activation=activation)
|
||||
X = np.random.randn(100,3)
|
||||
self.assertTrue(MappingGradChecker(mapping, X).checkgrad())
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue