Function main

examples/read-write/train_read-write.cc:83–122

}

int main(int argc, char** argv) {
  dynet::initialize(argc, argv);

  const unsigned HIDDEN = 8;
  const unsigned ITERATIONS = 20;
  ParameterCollection m;
  SimpleSGDTrainer trainer(m);
  XORModel model(HIDDEN, m);

  vector<dynet::real> x_values(2);  // set x_values to change the inputs
  dynet::real y_value;  // set y_value to change the target output

  // Train the model
  for (unsigned iter = 0; iter < ITERATIONS; ++iter) {
    double loss = 0;
    for (unsigned mi = 0; mi < 4; ++mi) {
      bool x1 = mi % 2;
      bool x2 = (mi / 2) % 2;
      x_values[0] = x1 ? 1 : -1;
      x_values[1] = x2 ? 1 : -1;
      y_value = (x1 != x2) ? 1 : -1;
      loss += model.Train(x_values, y_value, trainer);
    }
    loss /= 4;
    cerr << "E = " << loss << endl;
  }

  string outfile = "read-write.model";
  cerr << "Written model to File: " << outfile << endl;
  WriteToFile(outfile, m);  // Writing objects to file

  // New objects in which the written model will be read
  ParameterCollection read_dynet_model;
  XORModel read_model(HIDDEN, read_dynet_model);

  cerr << "Reading model from File: " << outfile << endl;
  ReadFromFile(outfile, read_dynet_model);  // Reading from file
  cerr << "Output for the input: " << x_values[0] << " " << x_values[1] << endl;
  cerr << read_model.Decode(x_values);  // Checking output for sanity
}