Copy code from cluster mixin
Use `transform` instead of `predict`
Cleanup
Add a simple test to compare to sklearn