summaryrefslogtreecommitdiff
path: root/model/ranksvmtn.h
diff options
context:
space:
mode:
Diffstat (limited to 'model/ranksvmtn.h')
-rw-r--r--model/ranksvmtn.h5
1 files changed, 2 insertions, 3 deletions
diff --git a/model/ranksvmtn.h b/model/ranksvmtn.h
index 4a0fb16..21b03bd 100644
--- a/model/ranksvmtn.h
+++ b/model/ranksvmtn.h
@@ -12,9 +12,8 @@ public:
{
return "TN";
};
-
- int train(DataProvider &D){return 0;};
- int predict(DataProvider &D){return 0;};
+ virtual int train(DataSet &D, Labels &label);
+ virtual int predict(DataSet &D, Eigen::MatrixXd &res);
};
#endif \ No newline at end of file