From 7d22423ac37f9362edc19e3318576e34e156eb73 Mon Sep 17 00:00:00 2001 From: abigailt Date: Wed, 31 May 2023 14:54:16 +0300 Subject: [PATCH] Reduce runtime of dataset assessment tests Signed-off-by: abigailt --- tests/test_data_assessment.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_data_assessment.py b/tests/test_data_assessment.py index 8ecad07..e4fe982 100644 --- a/tests/test_data_assessment.py +++ b/tests/test_data_assessment.py @@ -18,7 +18,7 @@ MIN_SHARE = 0.5 MIN_ROC_AUC = 0.0 MIN_PRECISION = 0.0 -NUM_SYNTH_SAMPLES = 40000 +NUM_SYNTH_SAMPLES = 400 NUM_SYNTH_COMPONENTS = 4 iris_dataset_np = get_iris_dataset_np() @@ -109,8 +109,8 @@ def kde(n_samples, n_components, original_data): digit_data = original_data pca = PCA(n_components=n_components, whiten=False) data = pca.fit_transform(digit_data) - params = {'bandwidth': np.logspace(-1, 1, 20)} - grid = GridSearchCV(KernelDensity(), params, cv=5) + params = {'bandwidth': np.logspace(-1, 1, 10)} + grid = GridSearchCV(KernelDensity(), params, cv=2) grid.fit(data) kde_estimator = grid.best_estimator_