Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

linear text dump model

Browse Source
This commit is contained in:
El Potaeto 2015-01-01 13:50:05 +01:00
parent d07be2bb96
commit 901904b535
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/gbm/gblinear-inl.hpp
View File

@ -8,6 +8,7 @@
*/ */
#include <vector> #include <vector>
#include <string> #include <string>
#include <sstream>
#include <algorithm> #include <algorithm>
#include "./gbm.h" #include "./gbm.h"
#include "../tree/updater.h" #include "../tree/updater.h"
@ -134,10 +135,23 @@ class GBLinear : public IGradBooster {
} }
} }
} }
virtual std::vector<std::string> DumpModel(const utils::FeatMap& fmap, int option) {
utils::Error("gblinear does not support dump model"); virtual std::vector<std::string> DumpModel(const utils::FeatMap& fmap, int option) {
return std::vector<std::string>(); std::stringstream fo("");
} fo << "bias:\n";
for (int i = 0; i < model.param.num_output_group; ++i) {
fo << model.bias()[i] << std::endl;
}
fo << "weight:\n";
for (int i = 0; i < model.param.num_output_group; ++i) {
for (int j = 0; j <model.param.num_feature; ++j) {
fo << model[i][j] << std::endl;
}
}
std::vector<std::string> v;
v.push_back(fo.str());
return v;
}
protected: protected:
inline void Pred(const RowBatch::Inst &inst, float *preds) { inline void Pred(const RowBatch::Inst &inst, float *preds) {