Skip to content

Instantly share code, notes, and snippets.

@crhea93
Created October 17, 2022 13:47
Show Gist options
  • Save crhea93/ffd278f7aa66d25ca92bb40c27beb586 to your computer and use it in GitHub Desktop.
Save crhea93/ffd278f7aa66d25ca92bb40c27beb586 to your computer and use it in GitHub Desktop.
test_dataset = tf.data.Dataset.from_tensor_slices((Y_test, A_test, N_test))
test_dataset = test_dataset.batch(batch_size, drop_remainder=True)
ysol = model(test_dataset)
# Obtain better format
ysol_list = []
for val in ysol:
ysol_ = [val.numpy() for val in val]
ysol_list.append(ysol_)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment