wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-01 00:35:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Shorten test_joblib (#8273)

Browse Source
This commit is contained in:
Edward Oakes
2020-05-01 17:11:32 -05:00
committed by GitHub
parent bf074073e7
commit d4e64709ba
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_joblib.py
+3 -5
View File
@@ -46,8 +46,7 @@ def test_svm_single_node(shutdown_only):
}
model = SVC(kernel="rbf")
search = RandomizedSearchCV(
model, param_space, cv=3, n_iter=50, verbose=10)
search = RandomizedSearchCV(model, param_space, cv=3, n_iter=2, verbose=10)
register_ray()
with joblib.parallel_backend("ray"):
search.fit(digits.data, digits.target)
@@ -64,8 +63,7 @@ def test_svm_multiple_nodes(ray_start_cluster_2_nodes):
}
model = SVC(kernel="rbf")
search = RandomizedSearchCV(
model, param_space, cv=5, n_iter=100, verbose=10)
search = RandomizedSearchCV(model, param_space, cv=5, n_iter=2, verbose=10)
register_ray()
with joblib.parallel_backend("ray"):
search.fit(digits.data, digits.target)
@@ -123,7 +121,7 @@ def test_sklearn_benchmarks(ray_start_cluster_2_nodes):
# Create train-test split.
print("Creating train-test split...")
n_train = 6000
n_train = 100
X_train = X[:n_train]
y_train = y[:n_train]
register_ray()