nubank / fklearn

fklearn: Functional Machine Learning
Apache License 2.0
1.51k stars 165 forks source link

Add K-means algorithm and unsupervised evaluators #219

Open fabiano-santos-nubank opened 1 year ago

fabiano-santos-nubank commented 1 year ago

Status

READY

Todo list

Background context

We are extending the fklearn capabilities with unsupervised algorithm support. First, we proposed to use k-means clustering implementation from scikit-learn since it is a simple and relevant unsupervised algorithm.

Description of the changes proposed in the pull request

Where should the reviewer start?

kmeans_learner method at src/fklearn/training/unsupervised.py. Then, the unsupervised evaluation metrics at src/fklearn/validation/evaluators.py.