enforce putting iteration numbers in train

2014-08-18 09:00:23 -07:00 · 2014-08-18 09:00:23 -07:00 · d3bfc31e6a
commit d3bfc31e6a
parent 3c1c7e2780
1 changed files with 6 additions and 2 deletions
--- a/python/xgboost.py
+++ b/python/xgboost.py
@ -134,8 +134,12 @@ class Booster:
                xglib.XGBoosterSetParam(
                    self.handle, ctypes.c_char_p(k.encode('utf-8')),
                    ctypes.c_char_p(str(v).encode('utf-8')))
-    def update(self, dtrain, it=1):
-        """ update """
+    def update(self, dtrain, it):
+        """ 
+        update 
+          dtrain: the training DMatrix
+          it: current iteration number
+        """        
        assert isinstance(dtrain, DMatrix)
        xglib.XGBoosterUpdateOneIter(self.handle, it, dtrain.handle)
    def boost(self, dtrain, grad, hess):