add entity_type node attr, support node attrs in dynamic Gephi export

Author: Aethor

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 @@ def _extract_graph(
 
         # * 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):