Fix `sklearn.cluster.KMeans` deprecation warning on Python 3.9 - Githubissues

icaros-usc / pyribs

A bare-bones Python library for quality diversity optimization.

https://pyribs.org

MIT License

210 stars 35 forks source link

Fix `sklearn.cluster.KMeans` deprecation warning on Python 3.9 #224

Closed itsdawei closed 2 years ago

itsdawei commented 2 years ago

Description

On python 3.9, sklearn.cluster.kmeans has deprecated algorithm='full'. According to the documentation for kmeans,

"auto" and "full" are deprecated and they will be removed in Scikit-Learn 1.3. They are both aliases for "lloyd".

This deprecation has led to a warning for some users, and this changes resolves that warning.

TODO

[x] Change algorithm='full' to algorithm='lloyd'

Status

[x] I have read the guidelines in CONTRIBUTING.md.
[x] I have formatted my code using yapf
[x] I have tested my code by running pytest
[x] I have linted my code with pylint
[x] I have added a one-line description of my change to the changelog in HISTORY.md.
- Exclude this from history
[x] This PR is ready to go