K-Means Clustering for Text

svtorykhsvtorykh MemberPosts:35Guru
edited December 2018 inHelp

Hi RM Team! I have a quck question about application of K-Means clustering for text.

I have a set of ~2000 comments. Once I'm done with Text Processing (using TF-IDF) I have a word vector matrix of ~30 terms.

I then apply K-means operator, but I wonder what actually serves as input for clustering? Is it vector matrix? If so, does clustering algorythm uses values from TF-IDF Word Vectors or some other values?

Best Answer

  • Telcontar120Telcontar120 Moderator, RapidMiner Certified Analyst, RapidMiner Certified Expert, MemberPosts:1,635Unicorn
    Solution Accepted

    Exactly, it is the word vector matrix that is used. So if you created the vector using TF-IDF, it will use those values. You also have the option of using other methods to create the vector like binary term occurrences or term frequency percentage.

    Brian T.
    Lindon Ventures
    Data Science Consulting from Certified RapidMiner Experts
    sgenzer

Answers

  • svtorykhsvtorykh MemberPosts:35Guru

    Thanks much!

    sgenzer
  • Telcontar120Telcontar120 Moderator, RapidMiner Certified Analyst, RapidMiner Certified Expert, MemberPosts:1,635Unicorn

    Your cluster will be based on the pruned values of the word vector. If you are interested in the details you should be able to review the actual values for each cluster on the centroid table output of the k-means operator.

    Brian T.
    Lindon Ventures
    Data Science Consulting from Certified RapidMiner Experts
    sgenzer
Sign InorRegisterto comment.