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')