From c8f422b3b9b15f3c93be90a00112f5ac738a8704 Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Wed, 24 Dec 2014 02:56:32 -0800
Subject: [PATCH] add dump to linear model

---
 src/gbm/gblinear-inl.hpp | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/gbm/gblinear-inl.hpp b/src/gbm/gblinear-inl.hpp
index 005eada55..8cbe8becf 100644
--- a/src/gbm/gblinear-inl.hpp
+++ b/src/gbm/gblinear-inl.hpp
@@ -8,6 +8,7 @@
  */
 #include <vector>
 #include <string>
+#include <sstream>
 #include <algorithm>
 #include "./gbm.h"
 #include "../tree/updater.h"
@@ -142,8 +143,20 @@ class GBLinear : public IGradBooster {
     utils::Error("gblinear does not support predict leaf index");
   }
   virtual std::vector<std::string> DumpModel(const utils::FeatMap& fmap, int option) {
-    utils::Error("gblinear does not support dump model");
-    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: