├── .gitignore ├── .isort.cfg ├── LICENSE ├── README.md ├── corener ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── entities.py │ └── sampling.py ├── evaluate.py ├── inference.py ├── models │ ├── __init__.py │ └── models.py ├── train.py └── utils │ ├── __init__.py │ ├── clusters.py │ ├── data.py │ ├── evaluation │ ├── __init__.py │ ├── coref_metrics.py │ └── evaluation.py │ ├── loss.py │ ├── model.py │ ├── prediction.py │ └── utils.py ├── examples └── inference_example.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- 1 | [settings] 2 | profile = black 3 | multi_line_output = 3 -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/README.md -------------------------------------------------------------------------------- /corener/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /corener/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/data/__init__.py -------------------------------------------------------------------------------- /corener/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/data/dataset.py -------------------------------------------------------------------------------- /corener/data/entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/data/entities.py -------------------------------------------------------------------------------- /corener/data/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/data/sampling.py -------------------------------------------------------------------------------- /corener/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/evaluate.py -------------------------------------------------------------------------------- /corener/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/inference.py -------------------------------------------------------------------------------- /corener/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/models/__init__.py -------------------------------------------------------------------------------- /corener/models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/models/models.py -------------------------------------------------------------------------------- /corener/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/train.py -------------------------------------------------------------------------------- /corener/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/__init__.py -------------------------------------------------------------------------------- /corener/utils/clusters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/clusters.py -------------------------------------------------------------------------------- /corener/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/data.py -------------------------------------------------------------------------------- /corener/utils/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/evaluation/__init__.py -------------------------------------------------------------------------------- /corener/utils/evaluation/coref_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/evaluation/coref_metrics.py -------------------------------------------------------------------------------- /corener/utils/evaluation/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/evaluation/evaluation.py -------------------------------------------------------------------------------- /corener/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/loss.py -------------------------------------------------------------------------------- /corener/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/model.py -------------------------------------------------------------------------------- /corener/utils/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/prediction.py -------------------------------------------------------------------------------- /corener/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/corener/utils/utils.py -------------------------------------------------------------------------------- /examples/inference_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/examples/inference_example.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aiola-lab/corener/HEAD/setup.py --------------------------------------------------------------------------------