add base_margin

2014-08-18 12:20:13 -07:00
parent 46fed899ab
commit 9da2ced8a2
12 changed files with 162 additions and 93 deletions
--- a/src/io/simple_dmatrix-inl.hpp
+++ b/src/io/simple_dmatrix-inl.hpp
@@ -110,10 +110,13 @@ class DMatrixSimple : public DataMatrix {
                   "DMatrix: group data does not match the number of rows in features");
    }
    std::string wname = name + ".weight";
-    if (info.TryLoadWeight(wname.c_str(), silent)) {
+    if (info.TryLoadFloatInfo("weight", wname.c_str(), silent)) {
      utils::Check(info.weights.size() == info.num_row,
                   "DMatrix: weight data does not match the number of rows in features");
    }
+    std::string mname = name + ".base_margin";
+    if (info.TryLoadFloatInfo("base_margin", mname.c_str(), silent)) {      
+    }
  }
  /*!
   * \brief load from binary file