Biaffine#

BiaffineDependencyParser#

class supar.models.dep.biaffine.BiaffineDependencyParser(*args, **kwargs)[source]#

The implementation of Biaffine Dependency Parser Dozat & Manning (2017).

MODEL#

alias of BiaffineDependencyModel

train(train: str | Iterable, dev: str | Iterable, test: str | Iterable, epochs: int = 1000, patience: int = 100, batch_size: int = 5000, update_steps: int = 1, buckets: int = 32, workers: int = 0, amp: bool = False, cache: bool = False, punct: bool = False, tree: bool = False, proj: bool = False, partial: bool = False, verbose: bool = True, **kwargs)[source]#
Parameters:
  • train/dev/test (Union[str, Iterable]) – Filenames of the train/dev/test datasets.

  • epochs (int) – The number of training iterations.

  • patience (int) – The number of consecutive iterations after which the training process would be early stopped if no improvement.

  • batch_size (int) – The number of tokens in each batch. Default: 5000.

  • update_steps (int) – Gradient accumulation steps. Default: 1.

  • buckets (int) – The number of buckets that sentences are assigned to. Default: 32.

  • workers (int) – The number of subprocesses used for data loading. 0 means only the main process. Default: 0.

  • clip (float) – Clips gradient of an iterable of parameters at specified value. Default: 5.0.

  • amp (bool) – Specifies whether to use automatic mixed precision. Default: False.

  • cache (bool) – If True, caches the data first, suggested for huge files (e.g., > 1M sentences). Default: False.

  • verbose (bool) – If True, increases the output verbosity. Default: True.

evaluate(data: str | Iterable, batch_size: int = 5000, buckets: int = 8, workers: int = 0, amp: bool = False, cache: bool = False, punct: bool = False, tree: bool = True, proj: bool = False, partial: bool = False, verbose: bool = True, **kwargs)[source]#
Parameters:
  • data (Union[str, Iterable]) – The data for evaluation. Both a filename and a list of instances are allowed.

  • batch_size (int) – The number of tokens in each batch. Default: 5000.

  • buckets (int) – The number of buckets that sentences are assigned to. Default: 8.

  • workers (int) – The number of subprocesses used for data loading. 0 means only the main process. Default: 0.

  • amp (bool) – Specifies whether to use automatic mixed precision. Default: False.

  • cache (bool) – If True, caches the data first, suggested for huge files (e.g., > 1M sentences). Default: False.

  • verbose (bool) – If True, increases the output verbosity. Default: True.

Returns:

The evaluation results.

predict(data: str | Iterable, pred: str | None = None, lang: str | None = None, prob: bool = False, batch_size: int = 5000, buckets: int = 8, workers: int = 0, amp: bool = False, cache: bool = False, tree: bool = True, proj: bool = False, verbose: bool = True, **kwargs)[source]#
Parameters:
  • data (Union[str, Iterable]) – The data for prediction. - a filename. If ends with .txt, the parser will seek to make predictions line by line from plain texts. - a list of instances.

  • pred (str) – If specified, the predicted results will be saved to the file. Default: None.

  • lang (str) – Language code (e.g., en) or language name (e.g., English) for the text to tokenize. None if tokenization is not required. Default: None.

  • prob (bool) – If True, outputs the probabilities. Default: False.

  • batch_size (int) – The number of tokens in each batch. Default: 5000.

  • buckets (int) – The number of buckets that sentences are assigned to. Default: 8.

  • workers (int) – The number of subprocesses used for data loading. 0 means only the main process. Default: 0.

  • amp (bool) – Specifies whether to use automatic mixed precision. Default: False.

  • cache (bool) – If True, caches the data first, suggested for huge files (e.g., > 1M sentences). Default: False.

  • verbose (bool) – If True, increases the output verbosity. Default: True.

Returns:

A Dataset object containing all predictions if cache=False, otherwise None.

classmethod build(path, min_freq=2, fix_len=20, **kwargs)[source]#

Build a brand-new Parser, including initialization of all data fields and model parameters.

Parameters:
  • path (str) – The path of the model to be saved.

  • min_freq (str) – The minimum frequency needed to include a token in the vocabulary. Required if taking words as encoder input. Default: 2.

  • fix_len (int) – The max length of all subword pieces. The excess part of each piece will be truncated. Required if using CharLSTM/BERT. Default: 20.

  • kwargs (Dict) – A dict holding the unconsumed arguments.

BiaffineDependencyModel#

class supar.models.dep.biaffine.BiaffineDependencyModel(n_words, n_rels, n_tags=None, n_chars=None, encoder='lstm', feat=['char'], n_embed=100, n_pretrained=100, n_feat_embed=100, n_char_embed=50, n_char_hidden=100, char_pad_index=0, elmo='original_5b', elmo_bos_eos=(True, False), bert=None, n_bert_layers=4, mix_dropout=0.0, bert_pooling='mean', bert_pad_index=0, finetune=False, n_plm_embed=0, embed_dropout=0.33, n_encoder_hidden=800, n_encoder_layers=3, encoder_dropout=0.33, n_arc_mlp=500, n_rel_mlp=100, mlp_dropout=0.33, scale=0, pad_index=0, unk_index=1, **kwargs)[source]#

The implementation of Biaffine Dependency Parser Dozat & Manning (2017).

Parameters:
  • n_words (int) – The size of the word vocabulary.

  • n_rels (int) – The number of labels in the treebank.

  • n_tags (int) – The number of POS tags, required if POS tag embeddings are used. Default: None.

  • n_chars (int) – The number of characters, required if character-level representations are used. Default: None.

  • encoder (str) – Encoder to use. 'lstm': BiLSTM encoder. 'bert': BERT-like pretrained language model (for finetuning), e.g., 'bert-base-cased'. Default: 'lstm'.

  • feat (List[str]) – Additional features to use, required if encoder='lstm'. 'tag': POS tag embeddings. 'char': Character-level representations extracted by CharLSTM. 'bert': BERT representations, other pretrained language models like RoBERTa are also feasible. Default: ['char'].

  • n_embed (int) – The size of word embeddings. Default: 100.

  • n_pretrained (int) – The size of pretrained word embeddings. Default: 100.

  • n_feat_embed (int) – The size of feature representations. Default: 100.

  • n_char_embed (int) – The size of character embeddings serving as inputs of CharLSTM, required if using CharLSTM. Default: 50.

  • n_char_hidden (int) – The size of hidden states of CharLSTM, required if using CharLSTM. Default: 100.

  • char_pad_index (int) – The index of the padding token in the character vocabulary, required if using CharLSTM. Default: 0.

  • elmo (str) – Name of the pretrained ELMo registered in ELMoEmbedding.OPTION. Default: 'original_5b'.

  • elmo_bos_eos (Tuple[bool]) – A tuple of two boolean values indicating whether to keep start/end boundaries of elmo outputs. Default: (True, False).

  • bert (str) – Specifies which kind of language model to use, e.g., 'bert-base-cased'. This is required if encoder='bert' or using BERT features. The full list can be found in transformers. Default: None.

  • n_bert_layers (int) – Specifies how many last layers to use, required if encoder='bert' or using BERT features. The final outputs would be weighted sum of the hidden states of these layers. Default: 4.

  • mix_dropout (float) – The dropout ratio of BERT layers, required if encoder='bert' or using BERT features. Default: .0.

  • bert_pooling (str) – Pooling way to get token embeddings. first: take the first subtoken. last: take the last subtoken. mean: take a mean over all. Default: mean.

  • bert_pad_index (int) – The index of the padding token in BERT vocabulary, required if encoder='bert' or using BERT features. Default: 0.

  • finetune (bool) – If False, freezes all parameters, required if using pretrained layers. Default: False.

  • n_plm_embed (int) – The size of PLM embeddings. If 0, uses the size of the pretrained embedding model. Default: 0.

  • embed_dropout (float) – The dropout ratio of input embeddings. Default: .33.

  • n_encoder_hidden (int) – The size of encoder hidden states. Default: 800.

  • n_encoder_layers (int) – The number of encoder layers. Default: 3.

  • encoder_dropout (float) – The dropout ratio of encoder layer. Default: .33.

  • n_arc_mlp (int) – Arc MLP size. Default: 500.

  • n_rel_mlp (int) – Label MLP size. Default: 100.

  • mlp_dropout (float) – The dropout ratio of MLP layers. Default: .33.

  • scale (float) – Scaling factor for affine scores. Default: 0.

  • pad_index (int) – The index of the padding token in the word vocabulary. Default: 0.

  • unk_index (int) – The index of the unknown token in the word vocabulary. Default: 1.

forward(words, feats=None)[source]#
Parameters:
  • words (LongTensor) – [batch_size, seq_len]. Word indices.

  • feats (List[LongTensor]) – A list of feat indices. The size is either [batch_size, seq_len, fix_len] if feat is 'char' or 'bert', or [batch_size, seq_len] otherwise. Default: None.

Returns:

The first tensor of shape [batch_size, seq_len, seq_len] holds scores of all possible arcs. The second of shape [batch_size, seq_len, seq_len, n_labels] holds scores of all possible labels on each arc.

Return type:

Tensor, Tensor

loss(s_arc, s_rel, arcs, rels, mask, partial=False)[source]#
Parameters:
  • s_arc (Tensor) – [batch_size, seq_len, seq_len]. Scores of all possible arcs.

  • s_rel (Tensor) – [batch_size, seq_len, seq_len, n_labels]. Scores of all possible labels on each arc.

  • arcs (LongTensor) – [batch_size, seq_len]. The tensor of gold-standard arcs.

  • rels (LongTensor) – [batch_size, seq_len]. The tensor of gold-standard labels.

  • mask (BoolTensor) – [batch_size, seq_len]. The mask for covering the unpadded tokens.

  • partial (bool) – True denotes the trees are partially annotated. Default: False.

Returns:

The training loss.

Return type:

Tensor

decode(s_arc, s_rel, mask, tree=False, proj=False)[source]#
Parameters:
  • s_arc (Tensor) – [batch_size, seq_len, seq_len]. Scores of all possible arcs.

  • s_rel (Tensor) – [batch_size, seq_len, seq_len, n_labels]. Scores of all possible labels on each arc.

  • mask (BoolTensor) – [batch_size, seq_len]. The mask for covering the unpadded tokens.

  • tree (bool) – If True, ensures to output well-formed trees. Default: False.

  • proj (bool) – If True, ensures to output projective trees. Default: False.

Returns:

Predicted arcs and labels of shape [batch_size, seq_len].

Return type:

LongTensor, LongTensor