compe_test_v1 / script.py
trtd56's picture
Update script.py
6334817 verified
raw
history blame
No virus
363 Bytes
import pandas as pd
print("################################")
from datasets import load_dataset
test_dset = load_dataset("ai-shift/compe_test", data_files={"test": "test.csv"})
print(test_dset)
print("################################")
df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])
df.to_csv("submission.csv", index=None)