NMF applied to Wikipedia articles
In the video, you saw NMF applied to transform a toy word-frequency array. Now it's your turn to apply NMF, this time using the tf-idf word-frequency array of Wikipedia articles, given as a csr matrix articles
. Here, fit the model and transform the articles. In the next exercise, you'll explore the result.
This exercise is part of the course
Unsupervised Learning in Python
Exercise instructions
- Import
NMF
fromsklearn.decomposition
. - Create an
NMF
instance calledmodel
with6
components. - Fit the model to the word count data
articles
. - Use the
.transform()
method ofmodel
to transformarticles
, and assign the result tonmf_features
. - Print
nmf_features
to get a first idea what it looks like (.round(2)
rounds the entries to 2 decimal places.)
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Import NMF
____
# Create an NMF instance: model
model = ____
# Fit the model to articles
____
# Transform the articles: nmf_features
nmf_features = ____
# Print the NMF features
print(nmf_features.round(2))