Skip to content

Instantly share code, notes, and snippets.

@ChunML
Last active December 12, 2018 05:40
Show Gist options
  • Save ChunML/d21071886bed3d0ef42c8f7c8348af98 to your computer and use it in GitHub Desktop.
Save ChunML/d21071886bed3d0ef42c8f7c8348af98 to your computer and use it in GitHub Desktop.
def reinitializable_input_fn(filenames, labels, train_val_ratio=0.8):
num_files = len(filenames)
num_train_files = int(num_files * train_val_ratio)
train_filenames = filenames[:num_train_files]
train_labels = labels[:num_train_files]
val_filenames = filenames[num_train_files:]
val_labels = labels[num_train_files:]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment