Installation¶

In [ ]:

# run this cell to install pycaret in Google Colab
# !pip install pycaret

In [1]:

# If you are using Jupyter notebook, you can pip install pycaret using jupyter notebook or command line
# pip install pycaret

In [2]:

from pycaret.utils import version
version()

1.0.0

In [ ]:

# only run this cell if you are using google colab
# from pycaret.utils import enable_colab
# enable_colab()

1. Importing Dataset¶

In [3]:

from pycaret.datasets import get_data
data = get_data('kiva')

	country	en	gender	loan_amount	nonpayment	sector
0	Dominican Republic	"Banco Esperanza" is a group of 10 women looki...	F	1225	partner	Retail
1	Dominican Republic	"Caminemos Hacia Adelante" or "Walking Forward...	F	1975	lender	Clothing
2	Dominican Republic	"Creciendo Por La Union" is a group of 10 peop...	F	2175	partner	Clothing
3	Dominican Republic	"Cristo Vive" ("Christ lives" is a group of 10...	F	1425	partner	Clothing
4	Dominican Republic	"Cristo Vive" is a large group of 35 people, 2...	F	4025	partner	Food

In [4]:

data.en[0]

Out[4]:

'"Banco Esperanza" is a group of 10 women looking to receive a small loan. Each of them has taken out a very small loan already, so this would be their second. With this loan the group is going to try and expand their small businesses and start generating more income. <P>\r\n\r\nEduviges is the group representative and leader of the group. Eduviges has a lot on the line because she has 6 children that she has to take care of. She told me that those children are the reason she wants to be successful. She wants to be able to provide a different life for them and show them that they can be successful as well. <P>\r\n\r\nEduviges has a very small business selling shoes and Avon products. She plans to expand using this loan and dreams of success. The whole group is ready for this new challenge and are on the road to bettering their lives. On behalf of Eduviges, the group, and Esperanza International--------- Thank you for your support!!!!'

In [5]:

data = data.head(1000)

2. Setting up Environment¶

In [6]:

from pycaret.nlp import *
nlp1 = setup(data, target='en', session_id=786)

Description	Value
session_id	786
# Documents	1000
Vocab Size	3870
Custom Stopwords	False

3. Train Topic Model¶

In [7]:

lda = create_model('lda')

In [8]:

print(lda)

LdaModel(num_terms=3870, num_topics=4, decay=0.5, chunksize=100)

In [9]:

nmf = create_model('nmf')

In [10]:

print(nmf)

NMF(alpha=0.0, beta_loss='frobenius', init='nndsvd', l1_ratio=0.0, max_iter=200,
    n_components=4, random_state=786, shuffle=False, solver='cd', tol=0.0001,
    verbose=0)

4. Analyze Trained Model¶

In [11]:

plot_model()

In [12]:

plot_model(lda)

In [13]:

plot_model(lda, plot = 'tsne')

In [14]:

plot_model(nmf, plot = 'tsne')

In [15]:

evaluate_model(lda)

interactive(children=(ToggleButtons(description='Plot Type:', icons=('',), options=(('Frequency Plot', 'freque…

5. Assign Labels / Predict on new data¶

In [16]:

results = assign_model(lda)
results.head()

Out[16]:

	country	en	gender	loan_amount	nonpayment	sector	Topic_0	Topic_1	Topic_2	Topic_3	Dominant_Topic	Perc_Dominant_Topic
0	Dominican Republic	group woman look receive small loan take small...	F	1225	partner	Retail	0.255099	0.114688	0.627855	0.002357	Topic 2	0.63
1	Dominican Republic	walk forward group entrepreneur seek second lo...	F	1975	lender	Clothing	0.570879	0.332298	0.094765	0.002057	Topic 0	0.57
2	Dominican Republic	group people hope start business group look re...	F	2175	partner	Clothing	0.165143	0.044879	0.786949	0.003029	Topic 2	0.79
3	Dominican Republic	live group woman look receive first loan young...	F	1425	partner	Clothing	0.324295	0.075030	0.598388	0.002287	Topic 2	0.60
4	Dominican Republic	vive large group people hope take loan many se...	F	4025	partner	Food	0.212557	0.152464	0.632609	0.002369	Topic 2	0.63

Learning Resources:¶

PyCaret NLP Module : https://www.pycaret.org/nlp
NLP Tutorial (Level Beginner) : https://www.pycaret.org/nlp101
NLP Tutorial (Level Intermediate) : https://www.pycaret.org/nlp102
Topic Modeling in PyCaret (Video Tutorial) : https://www.youtube.com/watch?v=G6ShuoM3T1M