From 9f32be3bdf475399124e18655e5458c84a349386 Mon Sep 17 00:00:00 2001 From: Harry Stuart Date: Tue, 11 Jun 2024 21:11:26 +1000 Subject: [PATCH] reduced benchmark execution time --- benchmarks/benchmark_performance.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/benchmarks/benchmark_performance.py b/benchmarks/benchmark_performance.py index e09ed87..da616ab 100644 --- a/benchmarks/benchmark_performance.py +++ b/benchmarks/benchmark_performance.py @@ -35,12 +35,12 @@ def mlflow_context(): @given( data=st.data(), feature_size=st.integers(min_value=1, max_value=10), - width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=200), - height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=200), - num_iterations=st.integers(min_value=10, max_value=1000), + width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20), + height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20), + num_iterations=st.integers(min_value=10, max_value=50), initial_learning_rate=st.floats(min_value=1e-3, max_value=1.0, allow_nan=False, allow_infinity=False) ) -@settings(max_examples=20, deadline=None, suppress_health_check=(HealthCheck.too_slow,)) +@settings(max_examples=10, deadline=None, suppress_health_check=(HealthCheck.too_slow,)) def benchmark_kohonen_networks_performance_mlflow(mlflow_context, data, feature_size, width, height, num_iterations, initial_learning_rate): np.random.seed(42) timestamp = datetime.datetime.now(datetime.UTC).strftime('%Y-%m-%dT%H:%M:%SZ')