test-submission / script.py
abhishek's picture
abhishek HF staff
Update script.py
2701b68
raw
history blame
No virus
308 Bytes
import pandas as pd
from transformers import AutoTokenizer
import os
print(os.system("ls -lahtr"))
tok = AutoTokenizer.from_pretrained("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)