Skip to content

Instantly share code, notes, and snippets.

@rnyak

rnyak/train_val_path

Last active July 23, 2020 13:51
Show Gist options
  • Save rnyak/cf3f20ff9e3391aedb31caafbdb7298b to your computer and use it in GitHub Desktop.
Save rnyak/cf3f20ff9e3391aedb31caafbdb7298b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_val_path
fname = 'day_{}.parquet'
NUM_TRAIN_DAYS = 23
num_days = len([i for i in os.listdir(INPUT_DATA_DIR) if re.match(fname.format('[0-9]{1,2}'), i) is not None])
train_path = [os.path.join(INPUT_DATA_DIR, fname.format(day)) for day in range(NUM_TRAIN_DAYS)]
valid_path = [os.path.join(INPUT_DATA_DIR, fname.format(day)) for day in range(NUM_TRAIN_DAYS, num_days)]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment