├── LICENSE ├── README.md ├── biaf_README.md ├── parser ├── __init__.py ├── biaf_evaluate.py ├── biaf_trainer.py ├── callbacks │ ├── __init__.py │ ├── eval_callback.py │ └── model_print_callback.py ├── data │ ├── __init__.py │ ├── base_bert_dataset.py │ ├── dependency_reader.py │ ├── s2s_dataset.py │ ├── sample.conllu │ ├── samplers │ │ ├── __init__.py │ │ └── grouped_sampler.py │ ├── span_proposal_dataset.py │ └── tree_utils.py ├── metrics │ ├── __init__.py │ ├── attachment_scores.py │ └── topk_acc.py ├── models │ ├── __init__.py │ ├── biaffine_dependency_config.py │ ├── biaffine_dependency_parser.py │ ├── s2s_query_dependency_config.py │ ├── s2s_query_dependency_parser.py │ ├── span_proposal.py │ └── span_proposal_config.py ├── s2s_evaluate_dp.py ├── s2s_query_trainer.py ├── span_proposal_trainer.py └── utils │ ├── __init__.py │ ├── decode_utils.py │ ├── get_parser.py │ └── logger.py ├── requirements.txt └── scripts ├── biaf ├── ctb │ ├── train_bert_chinese.sh │ └── train_roberta_chinese.sh ├── evaluate.sh ├── ptb │ ├── biaf_ptb.sh │ ├── ptb_finetune_bert.sh │ ├── train_bert.sh │ └── train_roberta.sh └── ud │ ├── train_bg.sh │ ├── train_ca.sh │ ├── train_cs.sh │ ├── train_de.sh │ ├── train_en.sh │ ├── train_es.sh │ ├── train_fr.sh │ ├── train_it.sh │ ├── train_nl.sh │ ├── train_no.sh │ ├── train_ro.sh │ └── train_ru.sh └── s2s ├── ctb5.1 ├── proposal.sh └── s2s.sh ├── ptb ├── proposal.sh └── s2s.sh ├── roberta_proposal.sh └── ud ├── bg ├── proposal.sh └── s2s.sh ├── ca ├── proposal.sh └── s2s.sh ├── cs ├── proposal.sh └── s2s.sh ├── de ├── proposal.sh └── s2s.sh ├── en ├── proposal.sh └── s2s.sh ├── es ├── proposal.sh └── s2s.sh ├── fr ├── proposal.sh └── s2s.sh ├── it ├── proposal.sh └── s2s.sh ├── nl ├── proposal.sh └── s2s.sh ├── no ├── proposal.sh └── s2s.sh ├── ro ├── proposal.sh └── s2s.sh └── ru ├── proposal.sh └── s2s.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/README.md -------------------------------------------------------------------------------- /biaf_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/biaf_README.md -------------------------------------------------------------------------------- /parser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parser/biaf_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/biaf_evaluate.py -------------------------------------------------------------------------------- /parser/biaf_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/biaf_trainer.py -------------------------------------------------------------------------------- /parser/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/callbacks/__init__.py -------------------------------------------------------------------------------- /parser/callbacks/eval_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/callbacks/eval_callback.py -------------------------------------------------------------------------------- /parser/callbacks/model_print_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/callbacks/model_print_callback.py -------------------------------------------------------------------------------- /parser/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parser/data/base_bert_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/base_bert_dataset.py -------------------------------------------------------------------------------- /parser/data/dependency_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/dependency_reader.py -------------------------------------------------------------------------------- /parser/data/s2s_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/s2s_dataset.py -------------------------------------------------------------------------------- /parser/data/sample.conllu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/sample.conllu -------------------------------------------------------------------------------- /parser/data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/samplers/__init__.py -------------------------------------------------------------------------------- /parser/data/samplers/grouped_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/samplers/grouped_sampler.py -------------------------------------------------------------------------------- /parser/data/span_proposal_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/span_proposal_dataset.py -------------------------------------------------------------------------------- /parser/data/tree_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/data/tree_utils.py -------------------------------------------------------------------------------- /parser/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/metrics/__init__.py -------------------------------------------------------------------------------- /parser/metrics/attachment_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/metrics/attachment_scores.py -------------------------------------------------------------------------------- /parser/metrics/topk_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/metrics/topk_acc.py -------------------------------------------------------------------------------- /parser/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/__init__.py -------------------------------------------------------------------------------- /parser/models/biaffine_dependency_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/biaffine_dependency_config.py -------------------------------------------------------------------------------- /parser/models/biaffine_dependency_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/biaffine_dependency_parser.py -------------------------------------------------------------------------------- /parser/models/s2s_query_dependency_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/s2s_query_dependency_config.py -------------------------------------------------------------------------------- /parser/models/s2s_query_dependency_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/s2s_query_dependency_parser.py -------------------------------------------------------------------------------- /parser/models/span_proposal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/span_proposal.py -------------------------------------------------------------------------------- /parser/models/span_proposal_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/models/span_proposal_config.py -------------------------------------------------------------------------------- /parser/s2s_evaluate_dp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/s2s_evaluate_dp.py -------------------------------------------------------------------------------- /parser/s2s_query_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/s2s_query_trainer.py -------------------------------------------------------------------------------- /parser/span_proposal_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/span_proposal_trainer.py -------------------------------------------------------------------------------- /parser/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parser/utils/decode_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/utils/decode_utils.py -------------------------------------------------------------------------------- /parser/utils/get_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/utils/get_parser.py -------------------------------------------------------------------------------- /parser/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/parser/utils/logger.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/biaf/ctb/train_bert_chinese.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ctb/train_bert_chinese.sh -------------------------------------------------------------------------------- /scripts/biaf/ctb/train_roberta_chinese.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ctb/train_roberta_chinese.sh -------------------------------------------------------------------------------- /scripts/biaf/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/evaluate.sh -------------------------------------------------------------------------------- /scripts/biaf/ptb/biaf_ptb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ptb/biaf_ptb.sh -------------------------------------------------------------------------------- /scripts/biaf/ptb/ptb_finetune_bert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ptb/ptb_finetune_bert.sh -------------------------------------------------------------------------------- /scripts/biaf/ptb/train_bert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ptb/train_bert.sh -------------------------------------------------------------------------------- /scripts/biaf/ptb/train_roberta.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ptb/train_roberta.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_bg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_bg.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_ca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_ca.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_cs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_cs.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_de.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_de.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_en.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_en.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_es.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_es.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_fr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_fr.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_it.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_it.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_nl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_nl.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_no.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_no.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_ro.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_ro.sh -------------------------------------------------------------------------------- /scripts/biaf/ud/train_ru.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/biaf/ud/train_ru.sh -------------------------------------------------------------------------------- /scripts/s2s/ctb5.1/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ctb5.1/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ctb5.1/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ctb5.1/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ptb/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ptb/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ptb/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ptb/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/roberta_proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/roberta_proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/bg/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/bg/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/bg/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/bg/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ca/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ca/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ca/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ca/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/cs/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/cs/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/cs/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/cs/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/de/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/de/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/de/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/de/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/en/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/en/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/en/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/en/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/es/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/es/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/es/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/es/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/fr/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/fr/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/fr/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/fr/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/it/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/it/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/it/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/it/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/nl/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/nl/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/nl/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/nl/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/no/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/no/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/no/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/no/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ro/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ro/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ro/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ro/s2s.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ru/proposal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ru/proposal.sh -------------------------------------------------------------------------------- /scripts/s2s/ud/ru/s2s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShannonAI/mrc-for-dependency-parsing/HEAD/scripts/s2s/ud/ru/s2s.sh --------------------------------------------------------------------------------