Harry/mantel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reduced benchmark execution time

Browse Source
This commit is contained in:
Harry Stuart 2024-06-11 21:11:26 +10:00
parent 75e019efc0
commit 9f32be3bdf
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
benchmarks/benchmark_performance.py
View File

@ -35,12 +35,12 @@ def mlflow_context():
@given( @given(
data=st.data(), data=st.data(),
feature_size=st.integers(min_value=1, max_value=10), feature_size=st.integers(min_value=1, max_value=10),
width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=200), width=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20),
height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=200), height=st.integers(min_value=MINIMUM_NETWORK_DIMENSION, max_value=20),
num_iterations=st.integers(min_value=10, max_value=1000), 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) 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): def benchmark_kohonen_networks_performance_mlflow(mlflow_context, data, feature_size, width, height, num_iterations, initial_learning_rate):
np.random.seed(42) np.random.seed(42)
timestamp = datetime.datetime.now(datetime.UTC).strftime('%Y-%m-%dT%H:%M:%SZ') timestamp = datetime.datetime.now(datetime.UTC).strftime('%Y-%m-%dT%H:%M:%SZ')