Search⌘ K
Home
Courses
Simplifying Machine Learning with PyCaret in Python

Solution Review: Plot Topic Distribution

Explore how to load data, set up a PyCaret NLP environment, build an LDA topic model, and plot topic distributions. Understand the step-by-step use of PyCaret for natural language processing tasks through practical application using the kiva dataset.

We'll cover the following...

Solution

Here is a sample solution for the challenge in the previous lesson.

Python 3.5
# Importing the necessary libraries
from pycaret.nlp import *
from pycaret.datasets import get_data
# Load the dataset
data = get_data('kiva')
# Get 1st 2000 instances of the dataset
data = data.iloc[0:2000]
# Initializing the PyCaret NLP environment
nlp_setup=setup(data = data, target = 'en', session_id = 123)
# Create the model
lda_model = create_model('lda')
# Plot the model
plot_model(lda_model, plot = 'topic_distribution')

Explanation

  • Lines 3–4: We import the necessary libraries ...