summaryrefslogtreecommitdiff
path: root/model/ranksvmtn.h
blob: 21b03bd6a6dc1d8a3c62d3e0796b35d56987b666 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#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, Eigen::MatrixXd &res);
};

#endif