def test_mlr_pickle(self):
tmp_file = mkstemp()[1]
model0 = MLR(10, 3)
model0.weights = randn(*model0.weights.shape)
model0.biases = randn(*model0.biases.shape)
# store model
with open(tmp_file, 'w') as handle:
dump({'model': model0}, handle)
# load model
with open(tmp_file) as handle:
model1 = load(handle)['model']
# make sure parameters haven't changed
self.assertEqual(model0.dim_in, model1.dim_in)
self.assertEqual(model0.dim_out, model1.dim_out)
self.assertLess(max(abs(model0.biases - model1.biases)), 1e-20)
self.assertLess(max(abs(model0.weights - model1.weights)), 1e-20)
def test_mlr_pickle(self):
tmp_file = mkstemp()[1]
model0 = MLR(10, 3)
model0.weights = randn(*model0.weights.shape)
model0.biases = randn(*model0.biases.shape)
# store model
with open(tmp_file, 'w') as handle:
dump({'model': model0}, handle)
# load model
with open(tmp_file) as handle:
model1 = load(handle)['model']
# make sure parameters haven't changed
self.assertEqual(model0.dim_in, model1.dim_in)
self.assertEqual(model0.dim_out, model1.dim_out)
self.assertLess(max(abs(model0.biases - model1.biases)), 1e-20)
self.assertLess(max(abs(model0.weights - model1.weights)), 1e-20)
