From e7abd0ab93402ab5db6a637b570a13872c8e2c39 Mon Sep 17 00:00:00 2001 From: peay Date: Mon, 1 Apr 2019 08:07:12 +0200 Subject: [PATCH] FIX Explicitly ignore SparseEfficiencyWarning in DBSCAN --- sklearn/cluster/dbscan_.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sklearn/cluster/dbscan_.py b/sklearn/cluster/dbscan_.py index f21beb3f91453..33ddf47948dc1 100644 --- a/sklearn/cluster/dbscan_.py +++ b/sklearn/cluster/dbscan_.py @@ -10,11 +10,11 @@ # License: BSD 3 clause import numpy as np +import warnings from scipy import sparse from ..base import BaseEstimator, ClusterMixin from ..utils import check_array, check_consistent_length -from ..utils.testing import ignore_warnings from ..neighbors import NearestNeighbors from ._dbscan_inner import dbscan_inner @@ -145,7 +145,8 @@ def dbscan(X, eps=0.5, min_samples=5, metric='minkowski', metric_params=None, X.sum_duplicates() # XXX: modifies X's internals in-place # set the diagonal to explicit values, as a point is its own neighbor - with ignore_warnings(): + with warnings.catch_warnings(): + warnings.simplefilter('ignore', sparse.SparseEfficiencyWarning) X.setdiag(X.diagonal()) # XXX: modifies X's internals in-place X_mask = X.data <= eps