Added mlp mapping (with possibility of having multiple layers).

This commit is contained in:
Neil Lawrence 2013-08-29 19:26:00 +02:00
parent e6739788ea
commit 128ebbabc5
12 changed files with 187 additions and 17 deletions

View file

@ -21,6 +21,12 @@ class MappingTests(unittest.TestCase):
self.assertTrue(GPy.core.Mapping_check_df_dtheta(mapping=mapping).checkgrad(verbose=verbose))
self.assertTrue(GPy.core.Mapping_check_df_dX(mapping=mapping).checkgrad(verbose=verbose))
def test_mlpmapping(self):
verbose = False
mapping = GPy.mappings.MLP(input_dim=2, hidden_dim=[3, 4, 8, 2], output_dim=2)
self.assertTrue(GPy.core.Mapping_check_df_dtheta(mapping=mapping).checkgrad(verbose=verbose))
self.assertTrue(GPy.core.Mapping_check_df_dX(mapping=mapping).checkgrad(verbose=verbose))
if __name__ == "__main__":