Added MNIST module and dataset

dainis-boumber · Nov 15, 2017 · a50eb3b · a50eb3b
1 parent 773738c
commit a50eb3b
Show file tree

Hide file tree

Showing 5 changed files with 84,168 additions and 4 deletions.
diff --git a/complexity.py b/complexity.py
@@ -7,6 +7,7 @@
 
 import modules.complexity_estimator as ce
 import modules.util as u
+from modules import mnist
 from modules.oracle import Oracle
 from nd_boundary_plot.plots import nd_boundary_plot
 
@@ -119,10 +120,11 @@ def main():
 
     # experiments.append('moons')
     # datasets.append((u.hastie(1000), u.hastie(1000)))
-    datasets.append((make_gaussian_quantiles(n_samples=500, n_features=10, n_classes=3),
-                     make_gaussian_quantiles(n_samples=500, n_features=10, n_classes=3)))
-    experiments.append('gauus')
-
+    # datasets.append((make_gaussian_quantiles(n_samples=500, n_features=10, n_classes=3),
+    #                 make_gaussian_quantiles(n_samples=500, n_features=10, n_classes=3)))
+    # experiments.append('gauus')
+    datasets.append((mnist.load_mnist(), mnist.load_mnist_rotated()))
+    experiments.append('MNIST_vs_MNIST_Rotated')
     active(classifiers=clfs, datasets=datasets, experiments=experiments)
 
 if __name__ == "__main__":