mxnet::cpp::EvalMetric Class Reference

#include <metric.h>

Inheritance diagram for mxnet::cpp::EvalMetric:

Collaboration diagram for mxnet::cpp::EvalMetric:

Public Member Functions
	EvalMetric (const std::string &name, int num=0)
virtual void	Update (NDArray labels, NDArray preds)=0
void	Reset ()
float	Get ()
void	GetNameValue ()

Static Protected Member Functions
static void	CheckLabelShapes (NDArray labels, NDArray preds, bool strict=false)

Protected Attributes
std::string	name
int	num
float	sum_metric = 0.0f
int	num_inst = 0

Constructor & Destructor Documentation

mxnet::cpp::EvalMetric::EvalMetric ( const std::string &  name, int  num = 0  )

inlineexplicit

Member Function Documentation

static void mxnet::cpp::EvalMetric::CheckLabelShapes ( NDArray  labels, NDArray  preds, bool  strict = false  )

inlinestaticprotected

float mxnet::cpp::EvalMetric::Get ( )

inline

void mxnet::cpp::EvalMetric::GetNameValue

(

)

void mxnet::cpp::EvalMetric::Reset ( )

inline

virtual void mxnet::cpp::EvalMetric::Update ( NDArray  labels, NDArray  preds  )

pure virtual

Implemented in mxnet::cpp::PSNR, mxnet::cpp::RMSE, mxnet::cpp::MSE, mxnet::cpp::MAE, mxnet::cpp::LogLoss, and mxnet::cpp::Accuracy.

Member Data Documentation

std::string mxnet::cpp::EvalMetric::name

protected

int mxnet::cpp::EvalMetric::num

protected

int mxnet::cpp::EvalMetric::num_inst = 0

protected

float mxnet::cpp::EvalMetric::sum_metric = 0.0f

protected

---

The documentation for this class was generated from the following file:

• cpp-package/include/mxnet-cpp/metric.h