metric.h

Go to the documentation of this file.

 
#ifndef NNLIB_METRIC_H
#define NNLIB_METRIC_H
 
#include "tensor.h"
 
class Metric {
 
protected:
    size_t numSamples;
 
    float currentTotalMetric;
 
public:
    Metric();
 
    void reset();
 
    virtual float calculateMetric(const sTensor& targets, const sTensor& predictions) = 0;
 
    [[nodiscard]] virtual std::string getShortName() const = 0;
};
 
class CategoricalAccuracy : public Metric {
 
public:
    CategoricalAccuracy();
 
    float calculateMetric(const sTensor& targets, const sTensor& predictions) override;
 
    [[nodiscard]] std::string getShortName() const override;
};
 
class BinaryAccuracy : public Metric {
 
public:
    BinaryAccuracy();
 
    float calculateMetric(const sTensor& targets, const sTensor& predictions) override;
 
    [[nodiscard]] std::string getShortName() const override;
};
 
#endif //NNLIB_METRIC_H