Update script.py
Browse files
script.py
CHANGED
|
@@ -30,8 +30,7 @@ if __name__ == "__main__":
|
|
| 30 |
solution.append({
|
| 31 |
'__key__': sample['__key__'],
|
| 32 |
'wf_vertices': pred_vertices.tolist(),
|
| 33 |
-
'wf_edges': pred_edges
|
| 34 |
-
'edge_semantics': semantics,
|
| 35 |
})
|
| 36 |
print('------------ Saving results ---------------')
|
| 37 |
sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges", "edge_semantics"])
|
|
|
|
| 30 |
solution.append({
|
| 31 |
'__key__': sample['__key__'],
|
| 32 |
'wf_vertices': pred_vertices.tolist(),
|
| 33 |
+
'wf_edges': pred_edges
|
|
|
|
| 34 |
})
|
| 35 |
print('------------ Saving results ---------------')
|
| 36 |
sub = pd.DataFrame(solution, columns=["__key__", "wf_vertices", "wf_edges", "edge_semantics"])
|