Update script.py

script.py

@@ -1,6 +1,8 @@
-import csv
+import pickle
+import numpy as np
+import pandas as pd
 
-with open('submission.csv', 'w', newline='') as file:
-    writer = csv.writer(file)
-    writer.writerow(['name', 'age'])
-    writer.writerow(['John Doe', 30])
+predict = [1,2,3,4]
+
+df = pd.DataFrame([(f"testid{i:04}", v) for i, v in enumerate(predict)], columns=["id", "pred"])
+df.to_csv("submission.csv", index=None)