completed argument mining demo notebooks

chrisPiemonte · Mar 8, 2020 · 1b9acb4 · 1b9acb4
1 parent fd96250
commit 1b9acb4
Show file tree

Hide file tree

Showing 4 changed files with 173 additions and 107 deletions.
diff --git a/notebooks/argument-mining/reddit.ipynb b/notebooks/argument-mining/reddit.ipynb
diff --git a/notebooks/argument-mining/stackoverflow.ipynb b/notebooks/argument-mining/stackoverflow.ipynb
diff --git a/notebooks/argument-mining/twitter.ipynb b/notebooks/argument-mining/twitter.ipynb
diff --git a/src/crawler/viz/drawer.py b/src/crawler/viz/drawer.py
@@ -29,12 +29,16 @@ def __init__(self, graph: Graph, **kwargs) -> None:
         self.options = options
 
 
-    def draw(self, ax: SubplotBase) -> None:
+    def draw(self, ax: SubplotBase, title: str = '') -> None:
         """[summary]
         
         Arguments:
             ax {SubplotBase} -- [description]
+        
+        Keyword Arguments:
+            title {str} -- [description] (default: {''})
         """
+        ax.set_title(title, fontsize=16)
         pos = nx.spring_layout(self.graph)
         if self.options['draw_edge_labels']:
             # self.options['edge_labels'] = nx.get_edge_attributes(self.graph, 'weight')