neural-network/CostFunction.h at main · connorkeevill/neural-network · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#pragma once

#include <vector>

using namespace std;

class CostFunction
{
public:
	virtual double Cost(vector<double> predicted, vector<double> expected) { return double{}; };
	virtual double Derivative(double predicted, double expected) { return double{}; };
	~CostFunction() = default;
protected:
	void guardAgainstInvalidVectorLengths(vector<double> a, vector<double> b);
};

class MeanSquaredError : public CostFunction
{
public:
	double Cost(vector<double> predicted, vector<double> expected) override;
	double Derivative(double predicted, double expected) override;
};

class CrossEntropy : public CostFunction
{
public:
    double Cost(vector<double> predicted, vector<double> expected) override;
    double Derivative(double predicted, double expected) override;
};