nlp_architect.models.cross_doc_coref package
Subpackages
Submodules
nlp_architect.models.cross_doc_coref.sieves_config module
-
class
nlp_architect.models.cross_doc_coref.sieves_config.
EntitySievesConfiguration
[source] Bases:
nlp_architect.models.cross_doc_coref.sieves_config.SievesConfiguration
-
class
nlp_architect.models.cross_doc_coref.sieves_config.
EventSievesConfiguration
[source] Bases:
nlp_architect.models.cross_doc_coref.sieves_config.SievesConfiguration
-
class
nlp_architect.models.cross_doc_coref.sieves_config.
SievesConfiguration
[source] Bases:
object
-
run_evaluation
Should run evaluation (True/False)
-
sieves_order
Sieve definition and Sieve running order
Tuple[SieveType, RelationType, Threshold(float)] - define sieves to run, were
Strict- Merge clusters only in case all mentions has current relation between them, Relax- Merge clusters in case (matched mentions) / len(cluster_1.mentions)) >= thresh, Very_Relax- Merge clusters in case (matched mentions) / (all possible pairs) >= thresh
RelationType represent the type of sieve to run.
-
nlp_architect.models.cross_doc_coref.sieves_resource module
-
class
nlp_architect.models.cross_doc_coref.sieves_resource.
SievesResources
[source] Bases:
object
-
elastic_host
Elastic host, #Required when using Elastic evaluation using Wikipedia sieve
-
elastic_index
Elastic index name, #Required when using Elastic evaluation using Wikipedia sieve
-
elastic_port
Elastic port number, #Required when using Elastic evaluation using Wikipedia sieve
-
elmo_file
Location of Elmo mini data set file, #Required mini data set file location for Offline evaluation using GloVe sieve
-
embed_search_method
EmbeddingMethod.GLOVE, EmbeddingMethod.GLOVE_OFFLINE, EmbeddingMethod.ELMO, EmbeddingMethod.ELMO_OFFLINE
Type: Wordnet search method type, one of
-
eval_output_dir
The output dir of the evaluation files, here scorer file for cross doc coref spans will be saved
-
glove_file
Location of GloVe mini data set file, #Required mini data set file location for Offline evaluation using GloVe sieve
-
referent_dict_file
Location of Referent dic data set file, #Required mini data set file for Offline evaluation or original file for Online evaluation using Referent Dict sieve
-
referent_dict_method
OnlineOROfflineMethod.ONLINE, OnlineOROfflineMethod.OFFLINE
Type: Referent Dict search method type, one of
-
vo_dict_file
Location of VerbOcean data set file, #Required mini data set file for Offline evaluation or original file for Online evaluation using VerbOcean sieve
-
vo_search_method
OnlineOROfflineMethod.ONLINE, OnlineOROfflineMethod.OFFLINE
Type: VerbOcean search method type, one of
-
wd_file
Location of Within doc data set file, #Required when using Within doc sieve
-
wiki_folder
Location of Wikipedia mini data set file, #Required mini data set file location for Offline evaluation using Wikipedia sieve
-
wiki_search_method
WikipediaSearchMethod.ONLINE, WikipediaSearchMethod.OFFLINE, WikipediaSearchMethod.ELASTIC
Type: Wikipedia search method type, one of
-
wn_folder
Location of WordNet mini data set file, #Required mini data set file location for Offline evaluation using WordNet sieve
-
wn_search_method
OnlineOROfflineMethod.ONLINE, OnlineOROfflineMethod.OFFLINE
Type: Wordnet search method type, one of
-