diff options
Diffstat (limited to 'model/ranksvmtn.cpp')
-rw-r--r-- | model/ranksvmtn.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/model/ranksvmtn.cpp b/model/ranksvmtn.cpp index cd58aa9..b82ce64 100644 --- a/model/ranksvmtn.cpp +++ b/model/ranksvmtn.cpp @@ -210,15 +210,16 @@ int RSVMTN::train(DataList &D){ vector<int> A1,A2; int i,j; LOG(INFO)<<"Processing input"; + vector<DataEntry*> &dat = D.getData(); for (i=0;i<D.getSize();++i) { - corr(i)=(D.getData()[i])->rank>0?0.5:-0.5; + corr(i)=(dat[i])->rank>0?0.5:-0.5; for (j = 0; j < D.getfSize(); ++j) - Data(i, j) = (D.getData()[i])->feature(j); + Data(i, j) = dat[i]->feature(j); } i=j=0; while (i<D.getSize()) { - if ((i+1 == D.getSize())|| D.getData()[i]->qid!=D.getData()[i+1]->qid) + if ((i+1 == D.getSize())|| dat[i]->qid!=dat[i+1]->qid) { A1.push_back(j); A2.push_back(i); |