|
13 | 13 |
|
14 | 14 |
|
15 | 15 | def get_parser(): |
16 | | - parser = argparse.ArgumentParser(description='Perform ARM, output mined rules as csv, get mined rules\' statistics') |
| 16 | + parser = argparse.ArgumentParser(prog='niaarm', |
| 17 | + description='Perform ARM, output mined rules as csv, get mined rules\' statistics') |
| 18 | + parser.add_argument('-i', '--input-file', type=str, required=True, help='Input file containing a csv dataset') |
| 19 | + parser.add_argument('-o', '--output-file', type=str, help='Output file for mined rules') |
17 | 20 | parser.add_argument('-a', '--algorithm', type=str, required=True, |
18 | | - help='Algorithm to use (niapy class name, e. g. DifferentialEvolution)') |
19 | | - parser.add_argument('-s', '--seed', type=int, help='Seed for thr algorithm\'s random number generator') |
| 21 | + help='Algorithm to use (niapy class name, e.g. DifferentialEvolution)') |
| 22 | + parser.add_argument('-s', '--seed', type=int, help='Seed for the algorithm\'s random number generator') |
20 | 23 | parser.add_argument('--max-evals', type=int, default=np.inf, help='Maximum number of fitness function evaluations') |
21 | 24 | parser.add_argument('--max-iters', type=int, default=np.inf, help='Maximum number of iterations') |
22 | | - parser.add_argument('-i', '--input-file', type=str, required=True, help='Input file containing a csv dataset') |
23 | | - parser.add_argument('-o', '--output-file', type=str, help='Output file for mined rules') |
24 | | - parser.add_argument('--alpha', type=float, default=0.0, help='Alpha parameter') |
25 | | - parser.add_argument('--beta', type=float, default=0.0, help='Beta parameter') |
26 | | - parser.add_argument('--gamma', type=float, default=0.0, help='Gamma parameter') |
27 | | - parser.add_argument('--delta', type=float, default=0.0, help='Delta parameter') |
| 25 | + parser.add_argument('--alpha', type=float, default=0.0, help='Alpha parameter. Default 0') |
| 26 | + parser.add_argument('--beta', type=float, default=0.0, help='Beta parameter. Default 0') |
| 27 | + parser.add_argument('--gamma', type=float, default=0.0, help='Gamma parameter. Default 0') |
| 28 | + parser.add_argument('--delta', type=float, default=0.0, help='Delta parameter. Default 0') |
28 | 29 | parser.add_argument('--logging', action='store_true', help='Enable logging of fitness improvements') |
29 | 30 | parser.add_argument('--show-stats', action='store_true', help='Display stats about mined rules') |
30 | 31 |
|
@@ -117,6 +118,7 @@ def main(): |
117 | 118 | algorithm.run(task) |
118 | 119 |
|
119 | 120 | if args.output_file: |
| 121 | + problem.sort_rules() |
120 | 122 | problem.export_rules(args.output_file) |
121 | 123 |
|
122 | 124 | if args.show_stats: |
|
0 commit comments