From 0c2c1c6aeaa7fa1acc1aa9c7bc34a1853fb6d522 Mon Sep 17 00:00:00 2001 From: shivesh Date: Sat, 13 Apr 2019 14:17:02 -0500 Subject: Sat 13 Apr 14:17:02 CDT 2019 --- plot_sens.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'plot_sens.py') 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') -- cgit v1.2.3