blob: 703fee47ed4e450624aca4cfb71efcdbf6057014 (
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(DataSet &D, Labels &label);
virtual int predict(DataSet &D, Labels &res);
};
int cg_solve(const Eigen::MatrixXd &A, const Eigen::VectorXd &b, Eigen::VectorXd &x);
#endif
|
