Allow custom mean-matching functions

AnotherSamWilson / miceforest

Multiple Imputation with LightGBM in Python

MIT License

353 stars 31 forks source link

Allow custom mean-matching functions #11

Closed samFarrellDay closed 3 years ago

samFarrellDay commented 3 years ago

Nearest neighbors search can be optimized for specific situations and machines. It would be a good idea to allow the user to specify a mean matching function if they don't want to rely on the default sklearn NearestNeighbors implementation. Over several million candidates, faiss is much faster.

AnotherSamWilson commented 3 years ago

Closed by 424c926eff8be2f590a0a816809ea5fb23da1998