[plotting] init figure_dict

plotting.py

@@ -129,10 +129,15 @@ def plot_histories(history_csv_path, path, cut=None, roll=1):
         plt.close("all")
 
 
-def plot_dict(importance, path="tmp.pdf"):
-    fig = plt.figure(figsize=(8, len(importance) / 3))
-    plt.barh(*zip(*importance.items()))
+def figure_dict(d):
+    fig = plt.figure(figsize=(8, len(d) / 3))
+    plt.barh(*zip(*d.items()))
     fig.tight_layout()
+    return fig
+
+
+def plot_dict(d, path="tmp.pdf"):
+    fig = figure_dict(d)
     plt.savefig(path)
     plt.close("all")