diff --git a/benchmarks/cb/preprocessing.py b/benchmarks/cb/preprocessing.py index 26b767f825..99b7737ed8 100644 --- a/benchmarks/cb/preprocessing.py +++ b/benchmarks/cb/preprocessing.py @@ -42,8 +42,8 @@ def apply_inplace_normalizer(X): def run_preprocessing_benchmarks(): - n_data_points = 20000 - n_features = 100 + n_data_points = 5000 + n_features = 50 X = ht.random.randn(n_data_points, n_features, split=0) apply_inplace_standard_scaler_and_inverse(X)