diff --git a/python/cuml/test/test_metrics.py b/python/cuml/test/test_metrics.py index 8d642b72fd..bb18ecddd4 100644 --- a/python/cuml/test/test_metrics.py +++ b/python/cuml/test/test_metrics.py @@ -152,7 +152,7 @@ def test_sklearn_search(): normalize=normalize, solver="eig") assert getattr(cu_clf, 'score', False) - sk_cu_grid = GridSearchCV(cu_clf, params, cv=5, iid=False) + sk_cu_grid = GridSearchCV(cu_clf, params, cv=5) gdf_data = cudf.DataFrame(X_train) gdf_train = cudf.DataFrame(dict(train=y_train)) diff --git a/python/cuml/test/test_pickle.py b/python/cuml/test/test_pickle.py index adf912f6f1..d961642a60 100644 --- a/python/cuml/test/test_pickle.py +++ b/python/cuml/test/test_pickle.py @@ -24,7 +24,7 @@ from cuml.test.test_svm import compare_svm, compare_probabilistic_svm from sklearn.base import clone from sklearn.datasets import load_iris, make_classification, make_regression -from sklearn.manifold.t_sne import trustworthiness +from sklearn.manifold import trustworthiness from sklearn.model_selection import train_test_split diff --git a/python/cuml/test/test_trustworthiness.py b/python/cuml/test/test_trustworthiness.py index 93819c1a4d..7a2a117067 100644 --- a/python/cuml/test/test_trustworthiness.py +++ b/python/cuml/test/test_trustworthiness.py @@ -13,7 +13,7 @@ # limitations under the License. import pytest -from sklearn.manifold.t_sne import trustworthiness as sklearn_trustworthiness +from sklearn.manifold import trustworthiness as sklearn_trustworthiness from cuml.metrics import trustworthiness as cuml_trustworthiness from sklearn.datasets import make_blobs diff --git a/python/cuml/test/test_tsne.py b/python/cuml/test/test_tsne.py index fcff0a49c5..8158543666 100644 --- a/python/cuml/test/test_tsne.py +++ b/python/cuml/test/test_tsne.py @@ -23,7 +23,7 @@ from cuml.neighbors import NearestNeighbors as cuKNN from sklearn.datasets import make_blobs -from sklearn.manifold.t_sne import trustworthiness +from sklearn.manifold import trustworthiness from sklearn import datasets diff --git a/python/cuml/test/test_umap.py b/python/cuml/test/test_umap.py index 5f8c54fc05..28b84e9375 100644 --- a/python/cuml/test/test_umap.py +++ b/python/cuml/test/test_umap.py @@ -37,7 +37,7 @@ from sklearn import datasets from sklearn.cluster import KMeans from sklearn.datasets import make_blobs -from sklearn.manifold.t_sne import trustworthiness +from sklearn.manifold import trustworthiness from sklearn.metrics import adjusted_rand_score dataset_names = ['iris', 'digits', 'wine', 'blobs']