salesforce / TransmogrifAI

TransmogrifAI (pronounced trăns-mŏgˈrə-fī) is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning
https://transmogrif.ai
BSD 3-Clause "New" or "Revised" License
2.24k stars 392 forks source link

how to use clustering algorithms ? #435

Closed rudyMcgee closed 4 years ago

rudyMcgee commented 4 years ago

I want to use clustering algorithms in my project,but I can't find anything about that

tovbinm commented 4 years ago

Hi! We have LDA already, which can be used as any other estimator in our examples. KMeans can be added in a similar fashion. Here is an example on how to use LDA - https://github.com/salesforce/TransmogrifAI/blob/master/core/src/test/scala/com/salesforce/op/stages/impl/feature/OpLDATest.scala

I hope this helps a bit.