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) |