fix by giulio

2014-08-25 12:10:45 -07:00
parent c78a2164c2
commit 6da62159d0
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 */