add possibility to save results from draw_pattern

skmine/periodic/cycles.py

@@ -416,7 +416,7 @@ def get_residuals(self, *patterns_id, sort="time"):
         return residuals_transf_pd
 
 
-    def draw_pattern(self, pattern_id):
+    def draw_pattern(self, pattern_id, directory=None):
         """
         Visually display a pattern based on its id from the discover command.
 
@@ -425,6 +425,9 @@ def draw_pattern(self, pattern_id):
         pattern_id : int
             The ID of the pattern to be displayed. This ID is to be retrieved directly from the discover command.
 
+        directory : str, default=None
+             Directory where the generated image and the DOT file are stored
+
         Returns
         -------
         Digraph
@@ -440,4 +443,8 @@ def draw_pattern(self, pattern_id):
             if isinstance(nid, int):
                 if "event" in pattern[nid].keys():
                     pattern[nid]["event"] = list(self.data_details.map_ev_num.keys())[int(pattern[nid]["event"])]
-        return draw_pattern(pattern)
+
+        graph = draw_pattern(pattern)
+        if directory:
+            graph.render(directory=directory)
+        return graph