[mlpack-git] [mlpack] Serialization of ann (#513)

Joseph Mariadassou notifications at github.com
Fri Mar 4 15:33:49 EST 2016


Serialization is not complete AFAIK: The following test fails (100% error):
```
  auto net = CreateVanillaNetwork();
  RMSprop<decltype(net)> opt(net, 0.01, 0.88, 1e-8, 10 * input.n_slices, 0);

  net.Train(input, Y, opt);
  data::Save("LearntMachine.xml", "ConvNetwork", net, mlpack::data::xml );
  size_t error = 0;
  {
    arma::mat prediction;
    net.Predict(input, prediction);
    ...
    }
  }
  auto net2 = CopyVanillaNetwork();
  data::Load("LearntMachine.xml", "ConvNetwork", net2, mlpack::data::xml);
  {
    arma::mat prediction;
    net2.Predict(input, prediction);
	...
}

```

Note that `net` computes the values as expected. But `net2` fails with 100% error. This is partly because the weights have not been serialized. But adding `network` and `outputLayer` to CNN::Serialize did not help.



---
Reply to this email directly or view it on GitHub:
https://github.com/mlpack/mlpack/issues/513#issuecomment-192457197
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160304/5c05a713/attachment.html>


More information about the mlpack-git mailing list