├── .gitignore ├── LICENSE ├── README.md ├── baselines ├── RSN.py ├── RSN_model.py ├── RoCORE_layers.py ├── RoCORE_model.py ├── etypeclus_model.py ├── latent_space_clustering.py ├── vae.py └── vqvae_model.py ├── common ├── __init__.py ├── b3.py ├── center_loss.py ├── clustering.py ├── convert_pred_for_eval.py ├── layers.py ├── log.py ├── metrics.py ├── predictions.py ├── sinkhorn_knopp.py └── utils.py ├── configs ├── event │ └── ace │ │ ├── pretrain.yaml │ │ ├── pretrain_e2e.yaml │ │ ├── spherical.yaml │ │ ├── ssvqvae.yaml │ │ ├── train.yaml │ │ └── train_e2e.yaml └── rel │ ├── fewrel │ ├── pretrain.yaml │ ├── rocore.yaml │ ├── rsn.yaml │ └── train.yaml │ └── tacred │ ├── pretrain.yaml │ ├── rocore.yaml │ ├── rsn.yaml │ └── train.yaml ├── data_sample ├── fewrel │ └── relation_description.csv └── tacred │ └── relation_description.csv ├── figures └── framework.jpg └── src ├── data_module.py ├── event_data_module.py ├── event_e2e_data_module.py ├── main.py ├── model.py └── multiview_layers.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/README.md -------------------------------------------------------------------------------- /baselines/RSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/RSN.py -------------------------------------------------------------------------------- /baselines/RSN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/RSN_model.py -------------------------------------------------------------------------------- /baselines/RoCORE_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/RoCORE_layers.py -------------------------------------------------------------------------------- /baselines/RoCORE_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/RoCORE_model.py -------------------------------------------------------------------------------- /baselines/etypeclus_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/etypeclus_model.py -------------------------------------------------------------------------------- /baselines/latent_space_clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/latent_space_clustering.py -------------------------------------------------------------------------------- /baselines/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/vae.py -------------------------------------------------------------------------------- /baselines/vqvae_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/baselines/vqvae_model.py -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/b3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/b3.py -------------------------------------------------------------------------------- /common/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/center_loss.py -------------------------------------------------------------------------------- /common/clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/clustering.py -------------------------------------------------------------------------------- /common/convert_pred_for_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/convert_pred_for_eval.py -------------------------------------------------------------------------------- /common/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/layers.py -------------------------------------------------------------------------------- /common/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/log.py -------------------------------------------------------------------------------- /common/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/metrics.py -------------------------------------------------------------------------------- /common/predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/predictions.py -------------------------------------------------------------------------------- /common/sinkhorn_knopp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/sinkhorn_knopp.py -------------------------------------------------------------------------------- /common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/common/utils.py -------------------------------------------------------------------------------- /configs/event/ace/pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/pretrain.yaml -------------------------------------------------------------------------------- /configs/event/ace/pretrain_e2e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/pretrain_e2e.yaml -------------------------------------------------------------------------------- /configs/event/ace/spherical.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/spherical.yaml -------------------------------------------------------------------------------- /configs/event/ace/ssvqvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/ssvqvae.yaml -------------------------------------------------------------------------------- /configs/event/ace/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/train.yaml -------------------------------------------------------------------------------- /configs/event/ace/train_e2e.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/event/ace/train_e2e.yaml -------------------------------------------------------------------------------- /configs/rel/fewrel/pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/fewrel/pretrain.yaml -------------------------------------------------------------------------------- /configs/rel/fewrel/rocore.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/fewrel/rocore.yaml -------------------------------------------------------------------------------- /configs/rel/fewrel/rsn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/fewrel/rsn.yaml -------------------------------------------------------------------------------- /configs/rel/fewrel/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/fewrel/train.yaml -------------------------------------------------------------------------------- /configs/rel/tacred/pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/tacred/pretrain.yaml -------------------------------------------------------------------------------- /configs/rel/tacred/rocore.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/tacred/rocore.yaml -------------------------------------------------------------------------------- /configs/rel/tacred/rsn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/tacred/rsn.yaml -------------------------------------------------------------------------------- /configs/rel/tacred/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/configs/rel/tacred/train.yaml -------------------------------------------------------------------------------- /data_sample/fewrel/relation_description.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/data_sample/fewrel/relation_description.csv -------------------------------------------------------------------------------- /data_sample/tacred/relation_description.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/data_sample/tacred/relation_description.csv -------------------------------------------------------------------------------- /figures/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/figures/framework.jpg -------------------------------------------------------------------------------- /src/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/data_module.py -------------------------------------------------------------------------------- /src/event_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/event_data_module.py -------------------------------------------------------------------------------- /src/event_e2e_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/event_e2e_data_module.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/main.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/model.py -------------------------------------------------------------------------------- /src/multiview_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raspberryice/type-discovery-abs/HEAD/src/multiview_layers.py --------------------------------------------------------------------------------