Initial commit for JSON.

dmlc · Jul 19, 2019 · 41b0bc2 · 41b0bc2
1 parent 1b3e869
commit 41b0bc2
Show file tree

Hide file tree

Showing 34 changed files with 3,702 additions and 25 deletions.
diff --git a/doc/json-schema.rst b/doc/json-schema.rst
diff --git a/include/xgboost/gbm.h b/include/xgboost/gbm.h
@@ -10,6 +10,7 @@
 
 #include <dmlc/registry.h>
 #include <xgboost/base.h>
+#include <xgboost/json.h>
 #include <xgboost/data.h>
 #include <xgboost/objective.h>
 #include <xgboost/feature_map.h>
@@ -46,11 +47,21 @@ class GradientBooster {
    * \param fi input stream.
    */
   virtual void Load(dmlc::Stream* fi) = 0;
+  /*!
+   * \brief load model from json
+   * \param in input json model.
+   */
+  virtual void Load(Json const& in) = 0;
   /*!
    * \brief save model to stream.
    * \param fo output stream
    */
   virtual void Save(dmlc::Stream* fo) const = 0;
+  /*!
+   * \brief Save model to Json
+   * \param out output json document.
+   */
+  virtual void Save(Json* out) const = 0;
   /*!
    * \brief whether the model allow lazy checkpoint
    * return true if model is only updated in DoBoost