Update script.py
Browse files
script.py
CHANGED
|
@@ -119,7 +119,7 @@ if __name__ == "__main__":
|
|
| 119 |
for i, sample in enumerate(tqdm(dataset[subset_name])):
|
| 120 |
res = process_sample(sample, i)
|
| 121 |
solution.append(res)
|
| 122 |
-
|
| 123 |
print('------------ Saving results ---------------')
|
| 124 |
sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
|
| 125 |
sub.to_parquet("submission.parquet")
|
|
|
|
| 119 |
for i, sample in enumerate(tqdm(dataset[subset_name])):
|
| 120 |
res = process_sample(sample, i)
|
| 121 |
solution.append(res)
|
| 122 |
+
|
| 123 |
print('------------ Saving results ---------------')
|
| 124 |
sub = pd.DataFrame(solution, columns=["order_id", "wf_vertices", "wf_edges"])
|
| 125 |
sub.to_parquet("submission.parquet")
|