A package to support running QNLP experiments involving anaphora. This package is meant to be used with discopy. An interactive demo can be found here.
pip install git+https://github.com/kinianlo/discopro.git --upgrade
Clone the repository with:
git clone https://github.com/kinianlo/discopro.git
Go into the repository directory:
cd discopro
Install the package with pip:
pip install . --upgrade
Consider a sentence that contains a anaphora, such as, Bob likes Alice. He was happy..
from discopy import Word, Ty, Id, Cup
from discopy.grammar import draw
# Define grammartical (pregroup) types
N = Ty('n') # noun
S = Ty('s') # sentence
# Define the words
bob = Word('Bob', N)
likes = Word('likes', N.r @ S @ N.l)
alice = Word('Alice', N)
he = Word('He', N)
was = Word('was', N.r @ S @ S.l @ N)
happy = Word('happy', N.r @ S)
# Define a discopy diagram
diagram = bob @ likes @ alice @ he @ was @ happy
diagram >>= Cup(N, N.r) @ Id(S) @ Cup(N.l, N) @ Cup(N, N.r) @ Id(S) @ diagram.cups(S.l @ N, N.r @ S)
draw(diagram)
One of the main usages of discopro is to connect the pronoun (He) and the referent (John) with a Cup(N, N.r):
# Connect the pronoun `He` to the referent `Bob`
from discopro.anaphora import connect_anaphora
ref_box_idx, pro_box_idx = 0, 3
diagram_connected = connect_anaphora(diagram, pro_box_idx, ref_box_idx)
draw(diagram_connected)
Note that some of the swaps introduced are unnecessary. One can use the min_swaps option in the connect_anaphora function to prevent unnecessary swaps.
diagram_connected = connect_anaphora(diagram, pro_box_idx, ref_box_idx, min_swaps=True)
draw(diagram_connected)
Another usage of discopro is to remove cups by flipping the word boxes. Sometimes swaps are also removed as a side effect.
from discopro.rewriting import contract
diagram_connected = contract(diagram_connected)
diagram_connected.draw()
Note that three cups were removed.
Swaps are preventing the algorithm from removing more cups. One can connect the anaphora with the referent on the top of the diagram without introducing any swaps in the first place.
from discopro.anaphora import connect_anaphora_on_top
diagram_connected = connect_anaphora_on_top(diagram, 3, 0)
diagram_connected.draw()
Note that doing so would render the diagram not being in the pregroup form. Now try to contract the connected diagram:
contract(diagram_connected).draw()
Notice that the word box for Bob also got flipped resulting in 2 more cups being removed.