22_train_classifier_ellipsoid_plane.cpp

This is an example of how to use the netdem library.

#include "general_net.hpp"
#include "mlpack_utils.hpp"
#include "regression_net.hpp"
 
using namespace netdem;
using namespace std;
 
void TrainClassifierEllipsoidPlane() {
  string root_dir = "local/examples/netdem/ann_models/ellipsoid_plane/";
 
  // notes: rows represent features, columns represent data points.
  arma::mat dataset;
  bool loadedDataset = mlpack::data::Load(
      root_dir + "archived/dataset_detection.txt", dataset, true);
  if (!loadedDataset) {
    cout << "loading data erros ..." << endl;
    return;
  }
  cout << "finished loading data ..." << endl;
 
  // split traning and testing data, prepare inputs and outputs
  arma::mat train_data, test_data;
  mlpack::data::Split(dataset, train_data, test_data, 0.1);
  int num_trains = train_data.n_cols;
  int num_tests = test_data.n_cols;
 
  int num_x{4}, num_y{1};
  arma::mat train_x = train_data.rows(0, num_x - 1);
  arma::mat test_x = test_data.rows(0, num_x - 1);
 
  arma::mat train_y = train_data.row(num_x);
  arma::mat test_y = test_data.row(num_x);
  cout << "finished preparing training and testing data ..." << endl;
 
  // create the ann model
  GeneralNet model;
  int num_nodes = 25, num_hidden_layers = 5;
  model.AddLayer(MLPackUtils::LayerType::Linear, num_nodes);
  model.AddLayer(MLPackUtils::LayerType::LeakyReLU);
  for (int i = 0; i < num_hidden_layers - 1; i++) {
    model.AddLayer(MLPackUtils::LayerType::Linear, num_nodes);
    model.AddLayer(MLPackUtils::LayerType::LeakyReLU);
  }
  model.AddLayer(MLPackUtils::LayerType::Linear, 2);
  model.AddLayer(MLPackUtils::LayerType::LogSoftMax);
  cout << "finished creating ann model ..." << endl;
 
  // set training parameters
  model.epochs = 100;
  model.step_size = 0.001;
  model.batch_size = 128;
  model.stop_tol = 1.0e-10;
 
  double test_accuracy_best{0};
  for (int i = 0; i < 10; i++) {
    model.Train(train_x, train_y);
    cout << endl << "finished training ..." << i << endl;
 
    // check the training and testing results
    auto pred_y = model.Classify(train_x);
    double train_accuracy =
        arma::accu(pred_y == train_y) * 1.0 / num_trains * 100.0;
    cout << "train accuracy: " << train_accuracy << " %" << endl;
 
    pred_y = model.Classify(test_x);
    double test_accuracy =
        arma::accu(pred_y == test_y) * 1.0 / num_tests * 100.0;
    cout << "test accuracy: " << test_accuracy << " %" << endl << endl;
 
    // save the model
    model.Save(root_dir + "training/ann_classifier_" + to_string(i) + ".xml",
               "detection");
 
    if (test_accuracy > test_accuracy_best) {
      test_accuracy_best = test_accuracy;
      model.Save(root_dir + "ann_classifier.xml", "detection");
    }
  }
}