diff options
| author | Joe Zhao <ztuowen@gmail.com> | 2015-04-11 17:46:19 +0800 |
|---|---|---|
| committer | Joe Zhao <ztuowen@gmail.com> | 2015-04-11 17:46:19 +0800 |
| commit | 85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e (patch) | |
| tree | 3059605ab0707b5530330dacdd9445b59bd73669 /model/ranksvm.cpp | |
| parent | 5d57accc3e1bc2b89c1e7753f7dbb40f3c8f575a (diff) | |
| download | ranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.tar.gz ranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.tar.bz2 ranksvm-85a4962556b67d1cc0668ecb2fbb03b3b4dd6e7e.zip | |
completed & tested, train & predict
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; |