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)