Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix benchmark_tree.py (#4593)

Browse Source
This commit is contained in:
Rong Ou 2019-06-20 16:51:48 -07:00 committed by Rory Mitchell
parent 9494950ee7
commit 77fc28427d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/benchmark/benchmark_tree.py
View File

@ -22,7 +22,7 @@ def run_benchmark(args):
if not (dtest.num_row() == args.rows * args.test_size if not (dtest.num_row() == args.rows * args.test_size
and dtrain.num_row() == args.rows * (1 - args.test_size)): and dtrain.num_row() == args.rows * (1 - args.test_size)):
raise ValueError("Wrong rows") raise ValueError("Wrong rows")
except ValueError: except:
print("Generating dataset: {} rows * {} columns".format(args.rows, args.columns)) print("Generating dataset: {} rows * {} columns".format(args.rows, args.columns))
print("{}/{} test/train split".format(args.test_size, 1.0 - args.test_size)) print("{}/{} test/train split".format(args.test_size, 1.0 - args.test_size))
tmp = time.time() tmp = time.time()