avoydatta / pybart

Converter from UD-trees to BART representation

A Python converter from Universal-Dependencies trees to BART representation.
Try out our UD-BART comparison Demo

---

BART (Bar-Ilan & AI2 Representation Transformation) is our new and cool enhanced-syntatic-representation specialized to improve Relation Extraction, but suitable for any NLP down-stream task.

See our pyBART: Evidence-based Syntactic Transformations for IE for detailed description of BART's creation/linguisical-verification/evaluation processes, and list of conversions.

This project is part of a wider project series, related to BART:

1. Converter: The current project.
2. Model: UD based spaCy model (pip install the_large_model). This model is needed when using the converter as a spaCy pipeline component (as spaCy doesn't provide UD-format based models).
3. Demo: Web-demo making use of the converter, to compare between UD and BART representations.

Table of contents

• Converter description
• Installation
• Usage
  • spaCy pipeline component
  • CoNLL-U format
• Configuration
• Citing
• Team

Table of contents generated with markdown-toc

Converter description

• Converts UD (v1.4) to BART.
• Supports Conll-U format, spaCy docs, and spaCy pipeline component (see Usage).
• Highly configurable (see Configuration).

Note: The BART representation subsumes Stanford's EnhancedUD conversions, these conversions are described here, and were already implemented by core-NLP Java converter. As such they were not avialable to python users and thus we have ported them to pyBART and tried to maintain their behavior as much as reasonable.

Installation

pyBART requires Python 3.7 or later. The preferred way to install pyBART is via pip. Just run pip install pybart-nlp in your Python environment and you're good to go! If you want to use pyBART as a spaCy pipeline component, then you should install as well: (1) the spaCy package and (2) a spaCy-model based on UD-format (which we happen to provide (details are here)

# if you want to use pyBART as a spaCy pipeline component, well,
#   you need spaCy installed and a spaCy model (based on UD-format):
pip install spacy
pip install https://storage.googleapis.com/en_ud_model/en_ud_model_lg-1.1.0.tar.gz

# or if you want smaller models:
#   medium: https://storage.googleapis.com/en_ud_model/en_ud_model_md-1.1.0.tar.gz
#   small: https://storage.googleapis.com/en_ud_model/en_ud_model_sm-1.1.0.tar.gz

# and this is us. please don't confuse with pybart/bart-py/bart
pip install pybart-nlp

Usage

Once you've installed pyBART, you can use the package in one of the following ways. Notice that for both methods the API calls can be called with a list of optional parameters to configure the conversion process. We will elaborate about them next.

spaCy pipeline component

import spacy

# Load a UD-based english model
nlp = spacy.load("en_ud_model_lg") # here you can change it to md/sm as you preffer

# Add BART converter to spaCy's pipeline
from pybart.api import Converter
converter = Converter()
nlp.add_pipe(converter, name="BART")

# Test the new converter component
doc = nlp("He saw me while driving")
me_token = doc[2]
for par_tok in me_token._.parent_list:
    print(par_tok)

# Output:
# {'head': saw, 'rel': 'dobj', 'src': 'UD', 'alt': None, 'unc': False}
# {'head': driving, 'rel': 'nsubj', 'src': ('advcl', 'while'), 'alt': 0, 'unc': False}

CoNLL-U format

from pybart.api import convert_bart_conllu

# read a CoNLL-U formatted file
with open(conllu_formatted_file_in) as f:
  sents = f.read()

# convert
converted = convert_bart_conllu(sents)

# use it, probably wanting to write the textual output to a new file
with open(conllu_formatted_file_out, "w") as f:
  f.write(converted)

Configuration

Each of our API calls can get the following optional parameters:

Name	Type	Default	Explanation
enhance_ud	boolean	True	Include Stanford's EnhancedUD conversions.
enhanced_plus_plus	boolean	True	Include Stanford's EnhancedUD++ conversions.
enhanced_extra	boolean	True	Include BART's unique conversions.
conv_iterations	int	inf	Stop the (defaultive) behaivor of iterating on the list of conversions after conv_iterations iterations, though before reaching convergance (that is, no change in graph when conversion-list is applied).
remove_eud_info	boolean	False	Do not include Stanford's EnhancedUD&EnhancedUD++'s extra label information.
remove_extra_info	boolean	False	Do not include BART's extra label information.
remove_node_adding_conversions	boolean	False	Do not include conversions that might add nodes to the given graph.
remove_unc	boolean	False	Do not include conversions that might contain uncertainty (see paper for detailed explanation).
query_mode	boolean	False	Do not include conversions that add arcs rather than reorder arcs.
funcs_to_cancel	ConvsCanceler class	Empty class instantiation	A list of conversions to prevent from occuring by their names. Use get_conversion_names for the full conversion name list

Citing

If you use pyBART or BART in your research, please cite pyBART: Evidence-based Syntactic Transformations for IE.

@inproceedings{Tiktinsky2020pyBARTES,
  title={pyBART: Evidence-based Syntactic Transformations for IE},
  author={Aryeh Tiktinsky and Yoav Goldberg and Reut Tsarfaty},
  year={2020}
}

Team

pyBART is an open-source project backed by the Allen Institute for Artificial Intelligence (AI2), and by Bar-Ilan University as being part of my thesis under the supervision of Yoav Goldberg. AI2 is a non-profit institute with the mission to contribute to humanity through high-impact AI research and engineering. Our team consists of Yoav Goldberg, Reut Tsarfaty and myself. Currently we are the contributors to this project but we will be more than happy for anyone who wants to help, via Issues and PR's.