forked from VowpalWabbit/vowpal_wabbit
/
loss_functions.h
39 lines (32 loc) · 1.27 KB
/
loss_functions.h
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
30
31
32
33
34
35
36
37
38
39
/*
Copyright (c) by respective owners including Yahoo!, Microsoft, and
individual contributors. All rights reserved. Released under a BSD
license as described in the file LICENSE.
*/
#ifndef LOSSFUNCTIONS_H_
#define LOSSFUNCTIONS_H_
#include <string>
#include "parse_primitives.h"
using namespace std;
class loss_function {
public :
/*
* getLoss evaluates the example loss.
* The function returns the loss value
*/
//virtual float getLoss(example *&ec, gd_vars &vars) = 0;
virtual float getLoss(shared_data*, float prediction, float label) = 0;
/*
* getUpdate evaluates the update scalar
* The function return the update scalar
*/
virtual float getUpdate(float prediction, float label, float eta_t, float norm) = 0;
virtual float getUnsafeUpdate(float prediction, float label, float eta_t, float norm) = 0;
virtual float getRevertingWeight(shared_data*, float prediction, float eta_t) = 0;
virtual float getSquareGrad(float prediction, float label) = 0;
virtual float first_derivative(shared_data*, float prediction, float label) = 0;
virtual float second_derivative(shared_data*, float prediction, float label) = 0;
virtual ~loss_function() {};
};
loss_function* getLossFunction(void*, string funcName, float function_parameter = 0);
#endif /* LOSSFUNCTIONS_H_ */