diff options
| author | shivesh <s.p.mandalia@qmul.ac.uk> | 2019-04-13 14:17:02 -0500 |
|---|---|---|
| committer | shivesh <s.p.mandalia@qmul.ac.uk> | 2019-04-13 14:17:02 -0500 |
| commit | 0c2c1c6aeaa7fa1acc1aa9c7bc34a1853fb6d522 (patch) | |
| tree | 028cdc45d1c0c66e56d02bb4566ba90933ada4ea /plot_sens.py | |
| parent | bb8f16faaaedae18e82049085c00920d3fa3a5f4 (diff) | |
| download | GolemFlavor-0c2c1c6aeaa7fa1acc1aa9c7bc34a1853fb6d522.tar.gz GolemFlavor-0c2c1c6aeaa7fa1acc1aa9c7bc34a1853fb6d522.zip | |
Sat 13 Apr 14:17:02 CDT 2019
Diffstat (limited to 'plot_sens.py')
| -rwxr-xr-x | plot_sens.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/plot_sens.py b/plot_sens.py index 0eef55e..f190905 100755 --- a/plot_sens.py +++ b/plot_sens.py @@ -27,6 +27,9 @@ from utils.misc import gen_identifier, SortingHelpFormatter from utils.param import Param, ParamSet +MASK_X = (0.3, 0.8) + + def process_args(args): """Process the input args.""" if args.data is not DataType.REAL: @@ -45,7 +48,11 @@ def process_args(args): args.source_ratios = map(fr_utils.normalise_fr, srs) elif args.x_segments is not None: x_array = np.linspace(0, 1, args.x_segments) - args.source_ratios = [[x, 1-x, 0] for x in x_array] + sources = [] + for x in x_array: + if x > MASK_X[0] and x < MASK_X[1]: continue + sources.append([x, 1-x, 0]) + args.source_ratios = sources else: raise ValueError('Must supply either --source-ratios or --x-segments') |