general_net.hpp

Go to the documentation of this file.

#pragma once
 
// netdem
#include "mlpack_utils.hpp"
 
// std
#include <cstdarg>
#include <string>
 
namespace netdem {
 
class GeneralNet {
public:
  mlpack::FFN<> 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 Classify(const arma::mat &data_x);
 
  arma::mat Regress(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