search

dmlc / MXNet.cpp

C++ interface for mxnet
Other
114 stars 78 forks source link

How to set different metric? #47

Closed horsepurve closed 7 years ago

horsepurve commented 8 years ago

In the examples I find no place to configure metric, so how can I specify different training metric e.g. acc, mse, auc, etc.?

Thanks! @thirdwing

thirdwing commented 8 years ago

I think right now we only have logloss and accuracy, see https://github.com/dmlc/MXNet.cpp/blob/master/include/mxnet-cpp/metric.h

You can write more if you need.