1. Learn
    2. /
  2. Courses
    3. /
  3. Feature Engineering for Machine Learning in Python

Exercise

Counting words (I)

Once high level information has been recorded you can begin creating features based on the actual content of each text. One way to do this is to approach it in a similar way to how you worked with categorical variables in the earlier lessons.

  • For each unique word in the dataset a column is created.
  • For each entry, the number of times this word occurs is counted and the count value is entered into the respective column.

These "count" columns can then be used to train machine learning models.

Instructions

100 XP
  • Import CountVectorizer from sklearn.feature_extraction.text.
  • Instantiate CountVectorizer and assign it to cv.
  • Fit the vectorizer to the text_clean column.
  • Print the feature names generated by the vectorizer.