fix

Author: piiswrong

README.md

@@ -2,7 +2,7 @@
 
 This package implements the algorithm described in paper "Unsupervised Deep Embedding for Clustering Analysis". It depends on opencv, numpy, scipy and Caffe.
 
-This implementation is intended for reproducing the results in the paper. If you only want to try the algorithm and find caffe too difficault to install, there is an experimental implementation in MXNet: https://github.com/dmlc/mxnet/blob/master/example/dec/dec.py. MXNet is a flexible deep learning library with fewer dependencies. You are welcome to try it. Installation guide can be found here: https://mxnet.readthedocs.org/en/latest/build.html. Once you install MXNet, simple go into directory examples/dec and run `python dec.py`.
+This implementation is intended for reproducing the results in the paper. If you only want to try the algorithm and find caffe too difficault to install, there is an easier to use experimental implementation in MXNet: https://github.com/dmlc/mxnet/blob/master/example/dec/dec.py, but note that results can be different from the paper. MXNet is a flexible deep learning library with fewer dependencies. You are welcome to try it. Installation guide can be found here: https://mxnet.readthedocs.org/en/latest/build.html. Once you install MXNet, simple go into directory examples/dec and run `python dec.py`.
 
 ## Usage
 To run, please first build our custom version of Caffe included in this package following the official guide: http://caffe.berkeleyvision.org/installation.html. 

dec/dec.py

@@ -747,7 +747,14 @@ def DisKmeans(db, update_interval = None):
 
 if __name__ == '__main__':
     db = sys.argv[1]
-    lam = 160
+    if db == 'mnist':
+        lam = 160
+    elif db == 'stl':
+        lam = 40
+    elif db == 'reutersidf' or db == 'reutersidf10k':
+        lam = 20
+    else:
+        lam = int(sys.argv[2])
     """acc_list = []
     nmi_list = []
     for i in xrange(0,9):