Loss.hpp

Go to the documentation of this file.

#ifndef LOSS_HPP_
#define LOSS_HPP_

#include <types/MatrixTypes.hpp>

namespace mic {
namespace neural_nets {
namespace loss {

template <typename dtype=float>
class Loss {
public:
    virtual dtype calculateLoss (mic::types::MatrixPtr<dtype> target_y_, mic::types::MatrixPtr<dtype> predicted_y_) = 0;

    virtual dtype calculateMeanLoss (mic::types::MatrixPtr<dtype> target_y_, mic::types::MatrixPtr<dtype> predicted_y_) {
        return calculateLoss(target_y_, predicted_y_) / predicted_y_->cols();
    }

    virtual mic::types::MatrixPtr<dtype> calculateGradient (mic::types::MatrixPtr<dtype> target_y_, mic::types::MatrixPtr<dtype> predicted_y_) = 0;

};

} //: loss
} //: neural_nets
} //: mic

#endif /* LOSS_HPP_ */