Merge branch 'unity' of ssh://github.com/tqchen/xgboost into unity

2014-08-25 13:57:21 -07:00
parent 3e9f8bfac9 e26af5e66c
commit 0066cd13a7
2 changed files with 4 additions and 2 deletions
--- a/src/tree/model.h
+++ b/src/tree/model.h
@@ -259,10 +259,12 @@ class TreeModel {
  }
  /*! \brief get leaf vector given nid */
  inline bst_float* leafvec(int nid) {
+    if (leaf_vector.size() == 0) return NULL;
    return &leaf_vector[nid * param.size_leaf_vector];
  }
  /*! \brief get leaf vector given nid */
  inline const bst_float* leafvec(int nid) const{
+    if (leaf_vector.size() == 0) return NULL;
    return &leaf_vector[nid * param.size_leaf_vector];
  }
  /*! \brief initialize the model */