loss.h

Go to the documentation of this file.

 
#ifndef NNLIB_LOSS_H
#define NNLIB_LOSS_H
 
#include "metric.h"
#include "tensor.h"
 
class Loss : public Metric {
 
public:
    Loss();
 
    float calculateMetric(const sTensor& targets, const sTensor& predictions) override;
 
    virtual sTensor calculateLoss(const sTensor& targets, const sTensor& predictions) = 0;
};
 
class MeanSquaredError : public Loss {
public:
    sTensor calculateLoss(const sTensor& targets, const sTensor& predictions) override;
 
    [[nodiscard]] std::string getShortName() const override;
};
 
class BinaryCrossEntropy : public Loss {
public:
    sTensor calculateLoss(const sTensor& targets, const sTensor& predictions) override;
 
    [[nodiscard]] std::string getShortName() const override;
};
 
class CategoricalCrossEntropy : public Loss {
public:
    sTensor calculateLoss(const sTensor& targets, const sTensor& predictions) override;
 
    [[nodiscard]] std::string getShortName() const override;
};
 
 
#endif //NNLIB_LOSS_H