blob: fd99d19d2e55e25bb4b4c7d38914a06b2729f7c9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#ifndef RSVMTN_H
#define RSVMTN_H
// Truncated Newton method based RankSVM
#include"ranksvm.h"
class RSVMTN:public RSVM
{
public:
std::string getName()
{
return "TN";
};
virtual int train(DataList &D);
virtual int predict(DataList &D,std::vector<double> &res);
};
int cg_solve(const Eigen::MatrixXd &A, const Eigen::VectorXd &b, Eigen::VectorXd &x);
#endif
|
