init version of lbfgs

2015-02-09 17:44:32 -08:00
parent 37a28376bb
commit 12ee049a74
9 changed files with 505 additions and 87 deletions
--- a/rabit-learn/kmeans/kmeans.cc
+++ b/rabit-learn/kmeans/kmeans.cc
@@ -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();