plots look good now

Ryanglambert · Mar 28, 2018 · 88289ab · 88289ab
1 parent c70a61c
commit 88289ab
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -14,6 +14,7 @@ ModelNet40/
 ModelNet40.zip.1
 ModelNet40.zip
 results_bak
+dont_track/*
 binvox
 viewvox
 *.log
diff --git a/ModelNet10_class_imbalance.png b/ModelNet10_class_imbalance.png
diff --git a/ModelNet40.png b/ModelNet40.png
diff --git a/ModelNet40_class_imbalance.png b/ModelNet40_class_imbalance.png
diff --git a/count_files.py b/count_files.py
@@ -32,14 +32,16 @@ def make_file_description_data_frame(path='./ModelNet10/', show_missed=True):
     return pd.DataFrame(paths)
 
 
-def plot_unbalanced_classes(df, title, display_df=False):
+def plot_unbalanced_classes(df, title, display_df=False, save=False):
     grouped = df.groupby(['class', 'type'])['type'].count()
     unstacked = grouped.unstack()
     plt.figure()
     unstacked = unstacked.sort_values(['test', 'train'])
     unstacked.plot(kind='bar', title=title, figsize=(10, 6), yticks=np.arange(0, unstacked['train'].max(), 100))
     plt.ylabel('Number of Samples')
     # plt.grid()
+    if save:
+        plt.savefig(title + '.png')
     plt.show()
     if display_df:
         display(unstacked)