Merge commit '13776a006a4e572720ec4c5b029b54771cf2b35c' into unity

subtree/rabit/rabit-learn/kmeans/kmeans.cc

@@ -2,8 +2,8 @@
 // facing an exception
 #include <rabit.h>
 #include <rabit/utils.h>
-#include "./toolkit_util.h"
 #include <time.h>
+#include "../utils/data.h"
 
 using namespace rabit;
 
@@ -83,9 +83,12 @@ inline size_t GetCluster(const Matrix &centroids,
              
 int main(int argc, char *argv[]) {
   if (argc < 5) {
+    // intialize rabit engine
+    rabit::Init(argc, argv);
     if (rabit::GetRank() == 0) {
       rabit::TrackerPrintf("Usage: <data_dir> num_cluster max_iter <out_model>\n");
     }
+    rabit::Finalize();
     return 0;
   }
   clock_t tStart = clock();