helper function to get q3 data separately, if students want to train …

…models independently.
bipiniiith · Feb 24, 2020 · 223cc10 · 223cc10
1 parent c36ce58
commit 223cc10
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/deepul/hw2_helper.py b/deepul/hw2_helper.py
@@ -165,6 +165,12 @@ def visualize_q3_data():
     images = train_data[idxs].astype(np.float32) / 3.0 * 255.0
     show_samples(images, title=f'{name} Samples')
 
+def get_q3_data():
+    data_dir = get_data_dir(2)
+    train_data, test_data = load_pickled_data(join(data_dir, 'celeb.pkl'))
+    return train_data, test_data
+
+
 def q3_save_results(fn, part):
     data_dir = get_data_dir(2)
     train_data, test_data = load_pickled_data(join(data_dir, 'celeb.pkl'))