def test_get_predictions(self):
network = Undertest.get_from_model(
"{}/model.hdf5".format(self.__model_dir), self.__hyperparameters)
with open(self.__train_data, "rb") as file:
train_data = pickle.load(file)
self.assertEqual(
(11431, 1),
network.get_predictions(
train_data,
"{}/weights.hdf5".format(self.__weights_dir)).shape)
def test_save_model_and_weights(self):
model_filepath = "{}/{}".format(self.__model_dir, "model.hdf5")
weights_filepath = "{}/{}".format(self.__weights_dir, "weights.hdf5")
Undertest.save_model_and_weights(
model_filepath,
weights_filepath,
"{}/{}".format(self.__resource_tmp, "model_combined.hdf5"),
)
self.assertEqual(
(2, 20),
Undertest.get_from_model(model_filepath,
self.__hyperparameters).input_shape)
def test_get_from_model(self):
model_filepath = "{}/{}".format(self.__model_dir, "model.hdf5")
network = Undertest.get_from_model(model_filepath,
self.__hyperparameters)
self.assertEqual((2, 20), network.input_shape)
self.assertEqual("unknown", network.n_type)