1 files changed, 3 insertions, 3 deletions
diff --git a/model/ranksvmtn.cpp b/model/ranksvmtn.cpp
index 559723c..3c9808b 100644
--- a/model/ranksvmtn.cpp
+++ b/model/ranksvmtn.cpp
@@ -126,8 +126,8 @@ int RSVMTN::train(DataSet &D, Labels &label){
 };
 
 int RSVMTN::predict(DataSet &D, Labels &res){
-    res = model.weight * D;
-    for (int i=0;i<res.cols();++i)
-        res[i] = (res[i] + model.beta);
+    //TODO define A
+    MatrixXd A;
+    res = A*(D * model.weight);
     return 0;
 };
 \ No newline at end of file