1 files changed, 81 insertions, 10 deletions

diff --git a/vest/mr_vest_generate_mapper_input.cc b/vest/mr_vest_generate_mapper_input.cc
index c96a61e4..9e702e2f 100644
--- a/vest/mr_vest_generate_mapper_input.cc
+++ b/vest/mr_vest_generate_mapper_input.cc
@@ -1,5 +1,6 @@
 #include <iostream>
 #include <vector>
+#include <sstream>
 
 #include <boost/program_options.hpp>
 #include <boost/program_options/variables_map.hpp>
@@ -7,10 +8,70 @@
 #include "filelib.h"
 #include "weights.h"
 #include "line_optimizer.h"
+#include "hg.h"
+#include "hg_io.h"
 
 using namespace std;
 namespace po = boost::program_options;
 
+typedef SparseVector<double> Dir;
+
+typedef RandomNumberGenerator<boost::mt19937> RNG;
+RNG rng;
+
+struct oracle_directions {
+  string forest_repository;
+  unsigned dev_set_size;
+  vector<Dir> dirs; //best_to_hope_dirs
+  vector<int> fids;
+  string forest_file(unsigned i) const {
+    ostringstream o;
+    o << forest_repository << '/' << i << ".json.gz";
+    return o.str();
+  }
+
+  oracle_directions(string forest_repository,unsigned dev_set_size,vector<int> const& fids=vector<int>()): forest_repository(forest_repository),dev_set_size(dev_set_size),fids(fids) {
+    dirs.resize(dev_set_size);
+  }
+  Dir const& operator[](unsigned i) {
+    Dir &dir=dirs[i];
+    if (dir.empty()) {
+      ReadFile rf(forest_file(i));
+      Hypergraph hg;
+      HypergraphIO::ReadFromJSON(rf.stream(), &hg);
+      cerr<<"oracle: forest["<<i<<"] loaded: "<<hg.stats()<<endl;
+      //TODO: get hope/oracle from vlad.  random for now.
+      LineOptimizer::RandomUnitVector(fids,&dir,&rng);
+    }
+    return dir;
+  }
+
+};
+
+void compress_similar(vector<Dir> &dirs,double min_dist,ostream *log=&cerr,bool avg=true) {
+  if (min_dist<=0) return;
+  double max_s=1.-min_dist;
+  unsigned N=dirs.size();
+  for (int i=0;i<N;++i) {
+    for (int j=i+1;j<N;++j) {
+      double s=dirs[i].tanimoto_coef(dirs[j]);
+      if (s>max_s) {
+        if (log) *log << "Collapsing similar directions (T="<<s<<" > "<<max_s<<").  dirs["<<i<<"]="<<dirs[i]<<" dirs["<<j<<"]";
+        if (avg) {
+          dirs[i]+=dirs[j];
+          dirs[i]/=2.;
+          if (log) *log<<" averaged="<<dirs[i];
+        }
+        if (log) *log<<endl;
+        swap(dirs[j],dirs[--N]);
+      }
+    }
+  }
+  dirs.resize(N);
+}
+
+
+
 void InitCommandLine(int argc, char** argv, po::variables_map* conf) {
   po::options_description opts("Configuration options");
   opts.add_options()
@@ -19,6 +80,10 @@ void InitCommandLine(int argc, char** argv, po::variables_map* conf) {
         ("weights,w",po::value<string>(),"[REQD] Current feature weights file")
         ("optimize_feature,o",po::value<vector<string> >(), "Feature to optimize (if none specified, all weights listed in the weights file will be optimized)")
         ("random_directions,d",po::value<unsigned int>()->default_value(20),"Number of random directions to run the line optimizer in")
+    ("no_primary,n","don't use the primary (orthogonal each feature alone) directions")
+    ("oracle_directions,O",po::value<unsigned>()->default_value(0),"read the forests and choose this many directions based on heading toward a hope max (bleu+modelscore) translation.")
+    ("oracle_batch,b",po::value<unsigned>()->default_value(10),"to produce each oracle direction, sum the 'gradient' over this many sentences")
+    ("max_similarity,m",po::value<double>()->default_value(0),"remove directions that are too similar (Tanimoto coeff. less than (1-this)).  0 means don't filter, 1 means only 1 direction allowed?")
         ("help,h", "Help");
   po::options_description dcmdline_options;
   dcmdline_options.add(opts);
@@ -43,30 +108,36 @@ void InitCommandLine(int argc, char** argv, po::variables_map* conf) {
 }
 
 int main(int argc, char** argv) {
-  RandomNumberGenerator<boost::mt19937> rng;
   po::variables_map conf;
   InitCommandLine(argc, argv, &conf);
   Weights weights;
   vector<string> features;
   weights.InitFromFile(conf["weights"].as<string>(), &features);
-  const string forest_repository = conf["forest_repository"].as<string>();
-  assert(DirectoryExists(forest_repository));
+  vector<int> fids(features.size());
+  for (int i = 0; i < features.size(); ++i)
+    fids[i] = FD::Convert(features[i]);
+
+  oracle_directions od(conf["forest_repository"].as<string>()
+                       , conf["dev_set_size"].as<unsigned int>()
+                       , fids
+    );
+;
+  assert(DirectoryExists(od.forest_repository));
   SparseVector<double> origin;
   weights.InitSparseVector(&origin);
   if (conf.count("optimize_feature") > 0)
     features=conf["optimize_feature"].as<vector<string> >();
   vector<SparseVector<double> > axes;
-  vector<int> fids(features.size());
-  for (int i = 0; i < features.size(); ++i)
-    fids[i] = FD::Convert(features[i]);
   LineOptimizer::CreateOptimizationDirections(
      fids,
      conf["random_directions"].as<unsigned int>(),
      &rng,
-     &axes);
-  int dev_set_size = conf["dev_set_size"].as<unsigned int>();
-  for (int i = 0; i < dev_set_size; ++i)
+     &axes,
+     !conf.count("no_primary")
+    );
+  compress_similar(axes,conf["max_similarity"].as<double>());
+  for (int i = 0; i < od.dev_set_size; ++i)
     for (int j = 0; j < axes.size(); ++j)
-      cout << forest_repository << '/' << i << ".json.gz " << i << ' ' << origin << ' ' << axes[j] << endl;
+      cout << od.forest_file(i) <<" " << i << ' ' << origin << ' ' << axes[j] << endl;
   return 0;
 }