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