import pandas as pd | |
print("################################") | |
from huggingface_hub import hf_hub_download | |
hf_hub_download(repo_id="ai-shift/compe_test_data", filename="test.csv", local_dir="./") | |
test_df = pd.read_csv("test.csv") | |
print("################################") | |
df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"]) | |
df.to_csv("submission.csv", index=None) |