Creating and Assigning the Topic Model
Learn to create, assign, and plot the topic model in natural language processing tasks.
We'll cover the following
Creating the topic model
The create_model()
function can be used to easily train a topic model on the preprocessed dataset. In this case, we train a LDA model with five topics, one for each of the dataset classes. After doing that, we use the assign_model()
function that assigns the topic proportions to the dataset, as well as the dominant topic. Finally, we create a new dataset containing the topic proportions as well as the news article class, and print the first 10 instances.
Get hands-on with 1400+ tech skills courses.