test-submission / script.py
abhishek's picture
abhishek HF staff
Update script.py
cfb2d0d
raw
history blame
No virus
250 Bytes
import pandas as pd
from transformers import AutoTokenizer
tok = AutoTokenizer("bert-base-uncased")
sub = []
for i in range(10000):
sub.append((i, 0.5))
sub = pd.DataFrame(sub, columns=["id", "pred"])
sub.to_csv("submission.csv", index=False)