mxnet::cpp::Block Class Reference

#include <Predictor.h>

Public Member Functions
const std::map< std::string, NDArray > &	arg_map () const
const std::map< std::string, NDArray > &	aux_map () const
	Block ()
	Block (const std::string &symbol_file, const std::string &param_file)
const Symbol &	symbol () const
Symbol	symbol (const std::string &output_node) const
virtual	~Block ()

Private Member Functions
void	load_parameters (const std::string &param_file)

Private Attributes
std::map< std::string, NDArray >	arg_map_
std::map< std::string, NDArray >	aux_map_
Symbol	sym_

Detailed Description

Definition at line 26 of file Predictor.h.

Constructor & Destructor Documentation

mxnet::cpp::Block::Block

(

)

Definition at line 19 of file Predictor.cc.

{}

mxnet::cpp::Block::Block	(	const std::string &	symbol_file,
		const std::string &	param_file
	)

Definition at line 21 of file Predictor.cc.

References load_parameters(), and sym_.

                                                                          {
      // load the symbol
      sym_ = Symbol::Load(symbol_file);
      // load the parameters
      load_parameters(param_file);
    }

mxnet::cpp::Block::~Block ( )

virtual

Definition at line 28 of file Predictor.cc.

{}

Member Function Documentation

const std::map<std::string, NDArray>& mxnet::cpp::Block::arg_map ( ) const

inline

Definition at line 34 of file Predictor.h.

References arg_map_.

{ return arg_map_; }

const std::map<std::string, NDArray>& mxnet::cpp::Block::aux_map ( ) const

inline

Definition at line 35 of file Predictor.h.

References aux_map_.

{ return aux_map_; }

void mxnet::cpp::Block::load_parameters ( const std::string &  param_file )

private

Definition at line 30 of file Predictor.cc.

References arg_map_, aux_map_, isotrackApplyRegressor::k, and mergeVDriftHistosByStation::name.

Referenced by Block().

                                                           {
      std::map<std::string, NDArray> paramters;
      NDArray::Load(param_file, nullptr, &paramters);
      for (const auto& k : paramters) {
        if (k.first.substr(0, 4) == "aux:") {
          auto name = k.first.substr(4, k.first.size() - 4);
          aux_map_[name] = k.second;
        }
        if (k.first.substr(0, 4) == "arg:") {
          auto name = k.first.substr(4, k.first.size() - 4);
          arg_map_[name] = k.second;
        }
      }
    }

const Symbol& mxnet::cpp::Block::symbol ( ) const

inline

Definition at line 32 of file Predictor.h.

References sym_.

{ return sym_; }

Symbol mxnet::cpp::Block::symbol ( const std::string &  output_node ) const

inline

Definition at line 33 of file Predictor.h.

References sym_.

{ return sym_.GetInternals()[output_node]; }

Member Data Documentation

std::map<std::string, NDArray> mxnet::cpp::Block::arg_map_

private

Definition at line 43 of file Predictor.h.

Referenced by arg_map(), and load_parameters().

std::map<std::string, NDArray> mxnet::cpp::Block::aux_map_

private

Definition at line 45 of file Predictor.h.

Referenced by aux_map(), and load_parameters().

Symbol mxnet::cpp::Block::sym_

private

Definition at line 41 of file Predictor.h.

Referenced by Block(), and symbol().