add cutomized training
This commit is contained in:
parent
9c2bb12cd1
commit
5fb9376af4
@ -86,7 +86,6 @@ bst = xgb.train( param, dtrain, num_round, evallist )
|
|||||||
#
|
#
|
||||||
print 'start running example to used cutomized objective function'
|
print 'start running example to used cutomized objective function'
|
||||||
|
|
||||||
|
|
||||||
# note: set loss_type properly, loss_type=2 means the prediction will get logistic transformed
|
# note: set loss_type properly, loss_type=2 means the prediction will get logistic transformed
|
||||||
# in most case, we may want to set loss_type = 0, to get untransformed score to compute gradient
|
# in most case, we may want to set loss_type = 0, to get untransformed score to compute gradient
|
||||||
bst = param = {'bst:max_depth':2, 'bst:eta':1, 'silent':1, 'loss_type':2 }
|
bst = param = {'bst:max_depth':2, 'bst:eta':1, 'silent':1, 'loss_type':2 }
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user