You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemml.apache.org by na...@apache.org on 2017/07/05 21:52:07 UTC
systemml git commit: [MINOR] More updates to the perftest help message
Repository: systemml
Updated Branches:
refs/heads/master 33cb26ded -> 1e1d3727f
[MINOR] More updates to the perftest help message
Project: http://git-wip-us.apache.org/repos/asf/systemml/repo
Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/1e1d3727
Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/1e1d3727
Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/1e1d3727
Branch: refs/heads/master
Commit: 1e1d3727f4d88c9ef053d56da7aec640e0b88424
Parents: 33cb26d
Author: Nakul Jindal <na...@gmail.com>
Authored: Wed Jul 5 14:50:38 2017 -0700
Committer: Nakul Jindal <na...@gmail.com>
Committed: Wed Jul 5 14:50:38 2017 -0700
----------------------------------------------------------------------
scripts/perftest/python/run_perftest.py | 32 ++++++++++++++++------------
1 file changed, 18 insertions(+), 14 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/systemml/blob/1e1d3727/scripts/perftest/python/run_perftest.py
----------------------------------------------------------------------
diff --git a/scripts/perftest/python/run_perftest.py b/scripts/perftest/python/run_perftest.py
index 6b8b4bc..dcc52c8 100755
--- a/scripts/perftest/python/run_perftest.py
+++ b/scripts/perftest/python/run_perftest.py
@@ -262,28 +262,32 @@ if __name__ == '__main__':
# Remove duplicates algorithms and used as default inputs
all_algos = set(reduce(lambda x, y: x + y, ML_ALGO.values()))
+ # Families
+ all_families = ML_ALGO.keys()
+
# Argparse Module
cparser = argparse.ArgumentParser(description='SystemML Performance Test Script')
- cparser.add_argument('--family', help='specify class of algorithms (available : ' + ', '.join(ML_ALGO.keys()) + ')',
- metavar='', choices=ML_ALGO.keys(), nargs='+')
- cparser.add_argument('--algo', help='specify the type of algorithm to run '
- '(Overrides --family, available : ' + ', '.join(all_algos) + ')', metavar='',
+ cparser.add_argument('--family', help='space separated list of classes of algorithms '
+ '(available : ' + ', '.join(sorted(all_families)) + ')',
+ metavar='', choices=all_families, nargs='+')
+ cparser.add_argument('--algo', help='space separated list of algorithm to run '
+ '(Overrides --family, available : ' + ', '.join(sorted(all_algos)) + ')', metavar='',
choices=all_algos, nargs='+')
cparser.add_argument('--exec-type', default='singlenode', help='System-ML backend '
- '(e.g singlenode, spark-hybrid)', metavar='',
+ '(available : singlenode, spark-hybrid)', metavar='',
choices=default_execution_mode)
- cparser.add_argument('--mat-type', default=default_mat_type, help='type of matrix to generate '
- '(e.g dense or sparse)', metavar='', choices=default_mat_type,
+ cparser.add_argument('--mat-type', default=default_mat_type, help='space separated list of types of matrix to generate '
+ '(available : dense, sparse)', metavar='', choices=default_mat_type,
nargs='+')
- cparser.add_argument('--mat-shape', default=default_mat_shape, help='shape of matrix '
- 'to generate (e.g 10k_1k)', metavar='', nargs='+')
- cparser.add_argument('--temp-dir', default=default_temp_dir, help='specify temporary directory',
- metavar='')
- cparser.add_argument('--filename', default='perf_test', help='specify output file for the perf'
- ' metics', metavar='')
+ cparser.add_argument('--mat-shape', default=default_mat_shape, help='space separated list of shapes of matrices '
+ 'to generate (e.g 10k_1k, 20M_4k)', metavar='', nargs='+')
+ cparser.add_argument('--temp-dir', default=default_temp_dir, help='temporary directory '
+ 'where generated, training and prediction data is put', metavar='')
+ cparser.add_argument('--filename', default='perf_test', help='name of the output file for the perf'
+ ' metrics', metavar='')
cparser.add_argument('--mode', default=default_workload,
- help='specify type of workload to run (e.g data-gen, train, predict)',
+ help='space separated list of types of workloads to run (available: data-gen, train, predict)',
metavar='', choices=default_workload, nargs='+')
# Args is a namespace