Added center box.

KateBlueSky · KateBlueSky · commit 34a30c74177e · 2025-03-17T16:32:36.000-07:00
diff --git a/sklbench/datasets/__init__.py b/sklbench/datasets/__init__.py
@@ -67,6 +67,8 @@ def load_data(bench_case: BenchCase) -> Tuple[Dict, Dict]:
             generation_kwargs = get_bench_case_value(
                 bench_case, "data:generation_kwargs", dict()
             )
+            if 'center_box' in generation_kwargs:
+                generation_kwargs['center_box'] = (-1 * generation_kwargs['center_box'], generation_kwargs['center_box'])
             return load_sklearn_synthetic_data(
                 function_name=source,
                 input_kwargs=generation_kwargs,

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,8 @@ def load_data(bench_case: BenchCase) -> Tuple[Dict, Dict]:`
`67`	`67`	`generation_kwargs = get_bench_case_value(`
`68`	`68`	`bench_case, "data:generation_kwargs", dict()`
`69`	`69`	`)`
	`70`	`+ if 'center_box' in generation_kwargs:`
	`71`	`+ generation_kwargs['center_box'] = (-1 * generation_kwargs['center_box'], generation_kwargs['center_box'])`
`70`	`72`	`return load_sklearn_synthetic_data(`
`71`	`73`	`function_name=source,`
`72`	`74`	`input_kwargs=generation_kwargs,`