diff options
Diffstat (limited to 'model')
| -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);  | 
