renard-pipeline 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

renard/graph_utils.py

@@ -124,6 +124,15 @@ def dynamic_graph_to_gephi_graph(graphs: List[nx.Graph]) -> nx.Graph:
     G = nx.Graph()
 
     for H_i, H in enumerate(graphs):
+
+        # update node attributes
+        for u in H.nodes:
+            G.add_node(u)
+            for key, value in H.nodes[u].items():
+                # format : (VALUE, START, END)
+                series = G.nodes[u].get(key, [])
+                G.nodes[u][key] = series + [(value, H_i, H_i + 1)]
+
         for u, v in H.edges:
             # create a new edge between two characters if it doesn't
             # exist yet

renard/pipeline/graph_extraction.py

@@ -306,8 +306,11 @@ class CoOccurrencesGraphExtractor(PipelineStep):
 
         # * Construct graph from co-occurence matrix
         G = nx.Graph()
-        for character, _ in mentions:
-            G.add_node(character)
+        for character, mention in mentions:
+            # NOTE: we add an 'entity_type' attribute. This is useful
+            # when using the 'additional_ner_classes' option, to
+            # differentiate between different entity types.
+            G.add_node(character, entity_type=mention.tag)
 
         for i, (char1, mention1) in enumerate(mentions):
             for j, (char2, mention2) in enumerate(mentions):

{renard_pipeline-0.6.1.dist-info → renard_pipeline-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: renard-pipeline
-Version: 0.6.1
+Version: 0.6.2
 Summary: Relationships Extraction from NARrative Documents
 Home-page: https://github.com/CompNet/Renard
 License: GPL-3.0-only

{renard_pipeline-0.6.1.dist-info → renard_pipeline-0.6.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 renard/gender.py,sha256=HDtJQKOqIkV8F-Mxva95XFXWJoKRKckQ3fc93OBM6sw,102
-renard/graph_utils.py,sha256=EV0_56KtI3VOElCu7wxd2kL8QVPsOu7itE6wGJAJsNA,6073
+renard/graph_utils.py,sha256=fcxXXcpzOkUJNgps0zgdoWJNydwkWwJJLJ6OlViNxTA,6373
 renard/ner_utils.py,sha256=SFZoyJM6c2avE7-NDkCSzkx-O8ppzS00a8EyHt64iGI,11628
 renard/nltk_utils.py,sha256=mUJiwMrEDZV4Fla7WuMR-hA_OC2ZIwSXgW_0Ew18VSo,977
 renard/pipeline/__init__.py,sha256=8Yim2mmny8YGvM7N5-na5zK-C9UDxUb77K9ml-VirUA,35
@@ -8,7 +8,7 @@ renard/pipeline/characters_extraction.py,sha256=bMic8dtlYKUmAlTzQqDPraYy5VsGWoGk
 renard/pipeline/core.py,sha256=LILUIQZp9f3FzqjBocUS7dKzX7lHQQVdL29jyqU1UeY,27754
 renard/pipeline/corefs/__init__.py,sha256=9c9AaXBcRrDBf1jhTtJ7DyjOJhX_Zej3FjlcGak7MK8,44
 renard/pipeline/corefs/corefs.py,sha256=d47Sd8ekwhQQV6rQ0F9QyAX2GOTqUnkDUA-eKgMtMS4,11417
-renard/pipeline/graph_extraction.py,sha256=Ga3wfUW9tDtatcTv2taLrNky9jz2wUwZ8uzoXJoSVk8,22928
+renard/pipeline/graph_extraction.py,sha256=CsQTdTKj2JmePgiw37_yF0SHcGTRmp62-8Faouaoty4,23154
 renard/pipeline/ner/__init__.py,sha256=Dqxcf_EKhK1UwiCscZ3gGHInlcxJyvpR4o-ZCLEyV48,38
 renard/pipeline/ner/ner.py,sha256=8zUtaqaGNirfGFRyMpDzdqtO3abrRLyLtjmwnqBNwUI,9893
 renard/pipeline/ner/retrieval.py,sha256=JIU3fi0Q1gl_YGP6kYx6zC9xz4UN6gnqdVuzWVXzzyM,12853
@@ -33,7 +33,7 @@ renard/resources/pronouns/pronouns.py,sha256=YJ8hM6H8QHrF2Xx6O5blqc-Sqe1D1YFL0sR
 renard/resources/titles/__init__.py,sha256=Jcg4B7stsWiAaXbFgNl_L3ICtCQmFe9bo3YjdkVL50w,45
 renard/resources/titles/titles.py,sha256=GsFccVJuTkgDWiAqWZpFd2R9pGvFKQZBOk4RWWuWDkw,968
 renard/utils.py,sha256=WL6djr3iu5Kzo2Jq6qDllHXgvZcEnmqBxPkQf1drq7c,4072
-renard_pipeline-0.6.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-renard_pipeline-0.6.1.dist-info/METADATA,sha256=vijGA3DMBq0Tkn2SJxMKacOw8zI5Z4IDSmIBWBuMEuM,4374
-renard_pipeline-0.6.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-renard_pipeline-0.6.1.dist-info/RECORD,,
+renard_pipeline-0.6.2.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+renard_pipeline-0.6.2.dist-info/METADATA,sha256=RtmfdAq8MeMrekPY_6CHjwkQAesrvAAGCclA_jLxRaM,4374
+renard_pipeline-0.6.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+renard_pipeline-0.6.2.dist-info/RECORD,,