regression_net.hpp

Go to the documentation of this file.

#pragma once
 
// netdem
#include "mlpack_utils.hpp"
 
// std
#include <cstdarg>
#include <string>
 
namespace netdem {
 
class RegressionNet {
public:
  mlpack::FFN<mlpack::MeanSquaredError, mlpack::HeInitialization> model;
 
  double step_size{0.01};
 
  int batch_size{32};
 
  double decay_rate_moment{0.9};
 
  double decay_rate_norm{0.9};
 
  double gradient_init_param{1e-8};
 
  int epochs{100};
 
  double stop_tol{1e-8};
 
  bool enable_logging{true};
 
  void ResetModel();
 
  void AddLayer(MLPackUtils::LayerType layer_name, ...);
 
  void Train(const arma::mat &data_x, const arma::mat &data_y);
 
  arma::mat Predict(const arma::mat &data_x);
 
  arma::mat Gradient(const arma::mat &data_x);
 
  void Load(std::string const &filename, std::string const &label);
 
  void Save(std::string const &filename, std::string const &label);
};
 
} // namespace netdem