Merge pull request #1 from frederik-elwert/feature/graphml

Add GraphML export.

Author: davidmcclure

textplot/graphs.py

@@ -56,6 +56,16 @@ def write_gml(self, path):
         nx.readwrite.gml.write_gml(self.graph, path)
 
 
+    def write_graphml(self, path):
+
+        """
+        Write a GraphML file.
+
+        :param path: The file path.
+        """
+        nx.readwrite.graphml.write_graphml(self.graph, path)
+
+
 class Skimmer(Graph):
 
 
@@ -86,4 +96,6 @@ def build(self, matrix, skim_depth=10, d_weights=False):
                 if d_weights: weight = 1-weight
 
                 n2 = matrix.text.unstem(term)
-                self.graph.add_edge(n1, n2, weight=weight)
+                # nx does not handle numpy types well when writing graphml,
+                # so ensure that weight is a regular float.
+                self.graph.add_edge(n1, n2, weight=float(weight))