summaryrefslogtreecommitdiff
path: root/model/ranksvmtn.h
blob: 40747812f5823552379155ca493ee66e06fb8432 (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(DataList &D);
    virtual int predict(DataList &D,std::vector<double> &res);
};

#endif