├── .gitignore ├── .pre-commit-config.yaml ├── .style.yapf ├── README.md ├── conf ├── conf.yaml ├── conf_test.yaml ├── data │ ├── brat │ │ ├── ace04_ner.yaml │ │ ├── ace05_ner.yaml │ │ ├── genia_ner.yaml │ │ └── nne_ner.yaml │ └── claim.yaml ├── dataloader │ ├── README.md │ ├── constant_token.yaml │ ├── constant_token_64b.yaml │ ├── constant_token_big.yaml │ └── constant_token_small.yaml ├── datamodule │ ├── _base.yaml │ └── brat_ner.yaml ├── embedding │ ├── _claim.yaml │ ├── _en.yaml │ ├── en.yaml │ └── en_transformer.yaml ├── encoder │ ├── parsing.yaml │ ├── span_rel.yaml │ ├── span_rel_mix.yaml │ ├── span_rel_sep.yaml │ ├── span_rel_sep2.yaml │ ├── span_rel_sep_mix.yaml │ ├── span_rel_sep_noarc.yaml │ └── span_rel_sep_spanlabel.yaml ├── hydra │ └── job_logging │ │ ├── custom.yaml │ │ └── nofile.yaml ├── metric │ └── span.yaml ├── model │ ├── ace04.yaml │ ├── ace05.yaml │ ├── genia.yaml │ └── nne.yaml ├── optimize │ ├── parsing.yaml │ ├── parsing_finetune.yaml │ ├── parsing_finetune1.yaml │ ├── parsing_finetune_cosine.yaml │ └── seqvat.yaml ├── runner │ └── basic.yaml ├── task │ └── ner_onestage2_gonly.yaml └── trainer │ ├── callbacks │ ├── best_watcher.yaml │ ├── early_stopping.yaml │ ├── lr_monitor.yaml │ ├── progressbar.yaml │ ├── swa.yaml │ └── wandb.yaml │ ├── debug.yaml │ ├── logger │ └── wandb.yaml │ ├── test.yaml │ ├── train.yaml │ └── train_wo_wandb.yaml ├── data └── raw2brat.py ├── requirements.txt ├── src ├── __init__.py ├── data │ ├── __init__.py │ ├── datamodule.py │ ├── datamodule_more │ │ ├── __init__.py │ │ ├── _sentencizer.py │ │ └── brat.py │ └── sampler.py ├── encoders │ ├── __init__.py │ ├── base.py │ ├── lstm.py │ └── mux.py ├── models │ └── basic.py ├── modules │ ├── __init__.py │ ├── affine_scorer.py │ ├── blocks │ │ ├── __init__.py │ │ ├── affine.py │ │ ├── attention.py │ │ ├── dropout.py │ │ ├── mlp.py │ │ ├── scalar_mix.py │ │ ├── span.py │ │ └── variational_lstm.py │ ├── common.py │ ├── configuration_roformer.py │ ├── embeddings │ │ ├── __init__.py │ │ ├── adaptor.py │ │ ├── embedding.py │ │ ├── fastnlp_adaptor.py │ │ ├── flair_adaptor.py │ │ ├── partial_trainable_static_embedding.py │ │ ├── torch_adaptor.py │ │ ├── transformers_embedding.py │ │ └── untruncated_embedding.py │ ├── linearchaincrf.py │ ├── loss │ │ ├── parsing │ │ │ ├── __init__.py │ │ │ ├── _cyk.py │ │ │ ├── _eisner_satta.py │ │ │ ├── _eisner_satta_exp.py │ │ │ ├── _fn.py │ │ │ ├── const.py │ │ │ ├── dep.py │ │ │ ├── joint.py │ │ │ └── sparsemax.py │ │ └── sjl_multilabel.py │ └── modeling_roformer.py ├── my_typing.py ├── runners │ ├── __init__.py │ └── basic.py ├── task_specific │ ├── base.py │ └── ner_onestage2_gonly.py ├── torch_struct │ ├── __init__.py │ ├── alignment.py │ ├── autoregressive.py │ ├── cky.py │ ├── cky_crf.py │ ├── deptree.py │ ├── distributions.py │ ├── helpers.py │ ├── linearchain.py │ ├── rl.py │ ├── semimarkov.py │ └── semirings │ │ ├── __init__.py │ │ ├── checkpoint.py │ │ ├── fast_semirings.py │ │ ├── keops.py │ │ ├── sample.py │ │ ├── semirings.py │ │ ├── sparse_max.py │ │ └── test_semirings.py └── utils │ ├── __init__.py │ ├── _logger.py │ ├── alg.py │ ├── callback.py │ ├── cl.py │ ├── config.py │ ├── fn.py │ ├── metric.py │ ├── scheduler.py │ └── var_pool.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/.style.yapf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/README.md -------------------------------------------------------------------------------- /conf/conf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/conf.yaml -------------------------------------------------------------------------------- /conf/conf_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/conf_test.yaml -------------------------------------------------------------------------------- /conf/data/brat/ace04_ner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/data/brat/ace04_ner.yaml -------------------------------------------------------------------------------- /conf/data/brat/ace05_ner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/data/brat/ace05_ner.yaml -------------------------------------------------------------------------------- /conf/data/brat/genia_ner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/data/brat/genia_ner.yaml -------------------------------------------------------------------------------- /conf/data/brat/nne_ner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/data/brat/nne_ner.yaml -------------------------------------------------------------------------------- /conf/data/claim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/data/claim.yaml -------------------------------------------------------------------------------- /conf/dataloader/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/dataloader/README.md -------------------------------------------------------------------------------- /conf/dataloader/constant_token.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/dataloader/constant_token.yaml -------------------------------------------------------------------------------- /conf/dataloader/constant_token_64b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/dataloader/constant_token_64b.yaml -------------------------------------------------------------------------------- /conf/dataloader/constant_token_big.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/dataloader/constant_token_big.yaml -------------------------------------------------------------------------------- /conf/dataloader/constant_token_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/dataloader/constant_token_small.yaml -------------------------------------------------------------------------------- /conf/datamodule/_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/datamodule/_base.yaml -------------------------------------------------------------------------------- /conf/datamodule/brat_ner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/datamodule/brat_ner.yaml -------------------------------------------------------------------------------- /conf/embedding/_claim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/embedding/_claim.yaml -------------------------------------------------------------------------------- /conf/embedding/_en.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/embedding/_en.yaml -------------------------------------------------------------------------------- /conf/embedding/en.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/embedding/en.yaml -------------------------------------------------------------------------------- /conf/embedding/en_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/embedding/en_transformer.yaml -------------------------------------------------------------------------------- /conf/encoder/parsing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/parsing.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_mix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_mix.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_sep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_sep.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_sep2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_sep2.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_sep_mix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_sep_mix.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_sep_noarc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_sep_noarc.yaml -------------------------------------------------------------------------------- /conf/encoder/span_rel_sep_spanlabel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/encoder/span_rel_sep_spanlabel.yaml -------------------------------------------------------------------------------- /conf/hydra/job_logging/custom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/hydra/job_logging/custom.yaml -------------------------------------------------------------------------------- /conf/hydra/job_logging/nofile.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/hydra/job_logging/nofile.yaml -------------------------------------------------------------------------------- /conf/metric/span.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/metric/span.yaml -------------------------------------------------------------------------------- /conf/model/ace04.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/model/ace04.yaml -------------------------------------------------------------------------------- /conf/model/ace05.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/model/ace05.yaml -------------------------------------------------------------------------------- /conf/model/genia.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/model/genia.yaml -------------------------------------------------------------------------------- /conf/model/nne.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/model/nne.yaml -------------------------------------------------------------------------------- /conf/optimize/parsing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/optimize/parsing.yaml -------------------------------------------------------------------------------- /conf/optimize/parsing_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/optimize/parsing_finetune.yaml -------------------------------------------------------------------------------- /conf/optimize/parsing_finetune1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/optimize/parsing_finetune1.yaml -------------------------------------------------------------------------------- /conf/optimize/parsing_finetune_cosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/optimize/parsing_finetune_cosine.yaml -------------------------------------------------------------------------------- /conf/optimize/seqvat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/optimize/seqvat.yaml -------------------------------------------------------------------------------- /conf/runner/basic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/runner/basic.yaml -------------------------------------------------------------------------------- /conf/task/ner_onestage2_gonly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/task/ner_onestage2_gonly.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/best_watcher.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/best_watcher.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/lr_monitor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/lr_monitor.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/progressbar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/progressbar.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/swa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/swa.yaml -------------------------------------------------------------------------------- /conf/trainer/callbacks/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/callbacks/wandb.yaml -------------------------------------------------------------------------------- /conf/trainer/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/debug.yaml -------------------------------------------------------------------------------- /conf/trainer/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/logger/wandb.yaml -------------------------------------------------------------------------------- /conf/trainer/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/test.yaml -------------------------------------------------------------------------------- /conf/trainer/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/train.yaml -------------------------------------------------------------------------------- /conf/trainer/train_wo_wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/conf/trainer/train_wo_wandb.yaml -------------------------------------------------------------------------------- /data/raw2brat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/data/raw2brat.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/data/datamodule.py -------------------------------------------------------------------------------- /src/data/datamodule_more/__init__.py: -------------------------------------------------------------------------------- 1 | from .brat import BratNER 2 | -------------------------------------------------------------------------------- /src/data/datamodule_more/_sentencizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/data/datamodule_more/_sentencizer.py -------------------------------------------------------------------------------- /src/data/datamodule_more/brat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/data/datamodule_more/brat.py -------------------------------------------------------------------------------- /src/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/data/sampler.py -------------------------------------------------------------------------------- /src/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/encoders/__init__.py -------------------------------------------------------------------------------- /src/encoders/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/encoders/base.py -------------------------------------------------------------------------------- /src/encoders/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/encoders/lstm.py -------------------------------------------------------------------------------- /src/encoders/mux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/encoders/mux.py -------------------------------------------------------------------------------- /src/models/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/models/basic.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/__init__.py -------------------------------------------------------------------------------- /src/modules/affine_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/affine_scorer.py -------------------------------------------------------------------------------- /src/modules/blocks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/__init__.py -------------------------------------------------------------------------------- /src/modules/blocks/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/affine.py -------------------------------------------------------------------------------- /src/modules/blocks/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/attention.py -------------------------------------------------------------------------------- /src/modules/blocks/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/dropout.py -------------------------------------------------------------------------------- /src/modules/blocks/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/mlp.py -------------------------------------------------------------------------------- /src/modules/blocks/scalar_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/scalar_mix.py -------------------------------------------------------------------------------- /src/modules/blocks/span.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/span.py -------------------------------------------------------------------------------- /src/modules/blocks/variational_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/blocks/variational_lstm.py -------------------------------------------------------------------------------- /src/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/common.py -------------------------------------------------------------------------------- /src/modules/configuration_roformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/configuration_roformer.py -------------------------------------------------------------------------------- /src/modules/embeddings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/__init__.py -------------------------------------------------------------------------------- /src/modules/embeddings/adaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/adaptor.py -------------------------------------------------------------------------------- /src/modules/embeddings/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/embedding.py -------------------------------------------------------------------------------- /src/modules/embeddings/fastnlp_adaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/fastnlp_adaptor.py -------------------------------------------------------------------------------- /src/modules/embeddings/flair_adaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/flair_adaptor.py -------------------------------------------------------------------------------- /src/modules/embeddings/partial_trainable_static_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/partial_trainable_static_embedding.py -------------------------------------------------------------------------------- /src/modules/embeddings/torch_adaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/torch_adaptor.py -------------------------------------------------------------------------------- /src/modules/embeddings/transformers_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/transformers_embedding.py -------------------------------------------------------------------------------- /src/modules/embeddings/untruncated_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/embeddings/untruncated_embedding.py -------------------------------------------------------------------------------- /src/modules/linearchaincrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/linearchaincrf.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/__init__.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/_cyk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/_cyk.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/_eisner_satta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/_eisner_satta.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/_eisner_satta_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/_eisner_satta_exp.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/_fn.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/const.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/const.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/dep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/dep.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/joint.py -------------------------------------------------------------------------------- /src/modules/loss/parsing/sparsemax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/parsing/sparsemax.py -------------------------------------------------------------------------------- /src/modules/loss/sjl_multilabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/loss/sjl_multilabel.py -------------------------------------------------------------------------------- /src/modules/modeling_roformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/modules/modeling_roformer.py -------------------------------------------------------------------------------- /src/my_typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/my_typing.py -------------------------------------------------------------------------------- /src/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/runners/__init__.py -------------------------------------------------------------------------------- /src/runners/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/runners/basic.py -------------------------------------------------------------------------------- /src/task_specific/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/task_specific/base.py -------------------------------------------------------------------------------- /src/task_specific/ner_onestage2_gonly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/task_specific/ner_onestage2_gonly.py -------------------------------------------------------------------------------- /src/torch_struct/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/__init__.py -------------------------------------------------------------------------------- /src/torch_struct/alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/alignment.py -------------------------------------------------------------------------------- /src/torch_struct/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/autoregressive.py -------------------------------------------------------------------------------- /src/torch_struct/cky.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/cky.py -------------------------------------------------------------------------------- /src/torch_struct/cky_crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/cky_crf.py -------------------------------------------------------------------------------- /src/torch_struct/deptree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/deptree.py -------------------------------------------------------------------------------- /src/torch_struct/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/distributions.py -------------------------------------------------------------------------------- /src/torch_struct/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/helpers.py -------------------------------------------------------------------------------- /src/torch_struct/linearchain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/linearchain.py -------------------------------------------------------------------------------- /src/torch_struct/rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/rl.py -------------------------------------------------------------------------------- /src/torch_struct/semimarkov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semimarkov.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/__init__.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/checkpoint.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/fast_semirings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/fast_semirings.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/keops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/keops.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/sample.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/semirings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/semirings.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/sparse_max.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/sparse_max.py -------------------------------------------------------------------------------- /src/torch_struct/semirings/test_semirings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/torch_struct/semirings/test_semirings.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/_logger.py -------------------------------------------------------------------------------- /src/utils/alg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/alg.py -------------------------------------------------------------------------------- /src/utils/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/callback.py -------------------------------------------------------------------------------- /src/utils/cl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/cl.py -------------------------------------------------------------------------------- /src/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/config.py -------------------------------------------------------------------------------- /src/utils/fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/fn.py -------------------------------------------------------------------------------- /src/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/metric.py -------------------------------------------------------------------------------- /src/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/scheduler.py -------------------------------------------------------------------------------- /src/utils/var_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/src/utils/var_pool.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LouChao98/nner_as_parsing/HEAD/train.py --------------------------------------------------------------------------------