Esempio n. 1
0
    def test_fileio_model(self):
        # Save
        model_name = 'test_model.lea'
        self.model.save(filename=TEMP_PATH + model_name)

        # Load
        test_model = NeuralNetwork.load(filename=TEMP_PATH + model_name)

        assert self.model.params == test_model.params
Esempio n. 2
0
    def test_fileio_model(self):
        # Save
        model_name = 'test_model.lea'
        self.model.save(filename=TEMP_PATH+model_name)

        # Load
        test_model = NeuralNetwork.load(filename=TEMP_PATH+model_name)

        assert self.model.params == test_model.params