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

auto jobname

Browse Source
This commit is contained in:
tqchen 2015-01-10 21:21:39 -08:00
parent 43c129f431
commit 0100fdd18d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tracker/rabit_hadoop.py
View File

@ -47,7 +47,7 @@ parser.add_argument('-f', '--files', nargs = '*',
' because the two files are cached to running folder.'\ ' because the two files are cached to running folder.'\
' You may need this option to cache additional files.'\ ' You may need this option to cache additional files.'\
' You can also use it to manually cache files when auto_file_cache is off') ' You can also use it to manually cache files when auto_file_cache is off')
parser.add_argument('--jobname', help = 'customize jobname in tracker') parser.add_argument('--jobname', default='auto', help = 'customize jobname in tracker')
parser.add_argument('--timeout', default=600000000, type=int, parser.add_argument('--timeout', default=600000000, type=int,
help = 'timeout of each mapper job, automatically set to a very long time,'\ help = 'timeout of each mapper job, automatically set to a very long time,'\
'normally you do not need to set this ') 'normally you do not need to set this ')
@ -72,7 +72,7 @@ parser.add_argument('command', nargs='+',
help = 'command for rabit program') help = 'command for rabit program')
args = parser.parse_args() args = parser.parse_args()
if args.jobname is None: if args.jobname == 'auto':
args.jobname = ('Rabit(nworker=%d):' % args.nworker) + args.command[0].split('/')[-1]; args.jobname = ('Rabit(nworker=%d):' % args.nworker) + args.command[0].split('/')[-1];
def hadoop_streaming(nworker, worker_args): def hadoop_streaming(nworker, worker_args):