diff options
Diffstat (limited to 'model/ranksvm.cpp')
| -rw-r--r-- | model/ranksvm.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model/ranksvm.cpp b/model/ranksvm.cpp index dc2ad9f..7ee72ac 100644 --- a/model/ranksvm.cpp +++ b/model/ranksvm.cpp @@ -40,8 +40,8 @@ RSVM* RSVM::loadModel(const string fname){ } int RSVM::setModel(const SVMModel &model) { - if (model.weight.cols()!=fsize) - LOG(FATAL) << "Feature size mismatch: "<<fsize<<" "<<model.weight.cols(); + if (model.weight.rows()!=fsize) + LOG(FATAL) << "Feature size mismatch: "<<fsize<<" "<<model.weight.rows(); this->model.weight=model.weight; this->model.beta=model.beta; return 0; |