some cleanup
This commit is contained in:
parent
baed0d0f08
commit
3975bf1e62
@ -63,9 +63,7 @@ for l in open('agaricus.txt.train'):
|
|||||||
i += 1
|
i += 1
|
||||||
|
|
||||||
csr = scipy.sparse.csr_matrix( (dat, (row,col)) )
|
csr = scipy.sparse.csr_matrix( (dat, (row,col)) )
|
||||||
print 'haha'
|
|
||||||
dtrain = xgb.DMatrix( csr )
|
dtrain = xgb.DMatrix( csr )
|
||||||
print 'set label'
|
|
||||||
dtrain.set_label(labels)
|
dtrain.set_label(labels)
|
||||||
evallist = [(dtest,'eval'), (dtrain,'train')]
|
evallist = [(dtest,'eval'), (dtrain,'train')]
|
||||||
bst = xgb.train( param, dtrain, num_round, evallist )
|
bst = xgb.train( param, dtrain, num_round, evallist )
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user