human-eval / long.py
PedroMartelleto's picture
update2
a779468
raw
history blame
307 Bytes
import pandas as pd
import random
video_pairs = pd.read_csv('file_pairs.csv')[['file_name', 'vista_id', 'gem_id']].values.tolist()
random.seed(42)
random.shuffle(video_pairs)
count = 0
for file_name, vista_id, gem_id in video_pairs[:20]:
print(file_name)
if "LONG" in file_name:
count += 1