mxnet
Public Member Functions | List of all members
mxnet::cpp::RMSE Class Reference

#include <metric.h>

Inheritance diagram for mxnet::cpp::RMSE:
Inheritance graph
Collaboration diagram for mxnet::cpp::RMSE:
Collaboration graph

Public Member Functions

 RMSE ()
 
void Update (NDArray labels, NDArray preds) override
 
- Public Member Functions inherited from mxnet::cpp::EvalMetric
 EvalMetric (const std::string &name, int num=0)
 
void Reset ()
 
float Get ()
 
void GetNameValue ()
 

Additional Inherited Members

- Static Protected Member Functions inherited from mxnet::cpp::EvalMetric
static void CheckLabelShapes (NDArray labels, NDArray preds, bool strict=false)
 
- Protected Attributes inherited from mxnet::cpp::EvalMetric
std::string name
 
int num
 
float sum_metric = 0.0f
 
int num_inst = 0
 

Constructor & Destructor Documentation

mxnet::cpp::RMSE::RMSE ( )
inline

Member Function Documentation

void mxnet::cpp::RMSE::Update ( NDArray  labels,
NDArray  preds 
)
inlineoverridevirtual

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