diff --git a/postprocess-spade.py b/postprocess-spade.py index 49c1afb..2110dcc 100644 --- a/postprocess-spade.py +++ b/postprocess-spade.py @@ -29,9 +29,9 @@ def process(source, ud, spade): columns = [ Column("ID", ColumnType.ID), - Column("UP:PREDS", ColumnType.UP_PREDS), - Column("UP:DEPARGS", ColumnType.UP_DEPARGS), - Column("UP:SPANARGS", ColumnType.UP_SPANARGS) + Column("UP:PRED", ColumnType.UP_PREDS), + Column("UP:ARGHEADS", ColumnType.UP_DEPARGS), + Column("UP:ARGSPANS", ColumnType.UP_SPANARGS) ] tree = Tree(columns) @@ -102,4 +102,5 @@ def process(source, ud, spade): t1 = time.time() - logging.info(f'Total postprocess-spade time: {(t1 - t0):.2f} s, processed sentences: {counter}') \ No newline at end of file + logging.info(f'Total postprocess-spade time: {(t1 - t0):.2f} s, processed sentences: {counter}') +