diff options
Diffstat (limited to 'model/ranksvmtn.h')
| -rw-r--r-- | model/ranksvmtn.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model/ranksvmtn.h b/model/ranksvmtn.h index 703fee4..6ed6ad7 100644 --- a/model/ranksvmtn.h +++ b/model/ranksvmtn.h @@ -12,8 +12,8 @@ public: { return "TN"; }; - virtual int train(DataSet &D, Labels &label); - virtual int predict(DataSet &D, Labels &res); + virtual int train(DataList &D); + virtual int predict(DataList &D,std::list<double> &res); }; int cg_solve(const Eigen::MatrixXd &A, const Eigen::VectorXd &b, Eigen::VectorXd &x); |