예제 #1
0
    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)
예제 #2
0
파일: mlr_test.py 프로젝트: cajal/cmt
	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)