Python Neural_Network.dumpの例

コード例 #1

ファイル: test_neuralnetwork.py プロジェクト: UriShavit/CystEarlyStagePrediction

 def test_MODEL_loading(self):
     nn = Neural_Network(preprocessor=self.pp3D)
     model_path = os.path.join(self.tmp_dir3D.name, "my_model.hdf5")
     nn.dump(model_path)
     nn_new = Neural_Network(preprocessor=self.pp3D)
     nn_new.load(model_path)

コード例 #2

ファイル: test_neuralnetwork.py プロジェクト: UriShavit/CystEarlyStagePrediction

 def test_MODEL_storage(self):
     nn = Neural_Network(preprocessor=self.pp3D)
     model_path = os.path.join(self.tmp_dir3D.name, "my_model.hdf5")
     nn.dump(model_path)
     self.assertTrue(os.path.exists(model_path))