DistrictDataLabs / yellowbrick / 1047

90%

push

travis-ci

Updates the distortion_score to handle sparse matrices (#371)

We identified an issue finding the centroid of a sparse matrix
(scipy.sparse.csc_matrix and scipy.sparse.csr_matrix) -- the scipy mean
returns a 2D array instead of an array of 1 dimension less than the
input (like numpy does). I used sort of a naive check,
scipy.sparse.issparse to check the matrix then updated the array
correctly. Added tests for csr, csc, as well as data frames to
test_elbow.py

Fixes #370

5 of 5 new or added lines in 1 file covered. (100.0%)

2084 of 2506 relevant lines covered (83.16%)

1.66 hits per line