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)