Skip to content

Instantly share code, notes, and snippets.

@ecdedios

ecdedios/pycaret-topic-modeling-the-meat

Created August 10, 2022 04:03
Show Gist options
  • Save ecdedios/8e04644657f2a4fc61f6d400ec6a9881 to your computer and use it in GitHub Desktop.
Save ecdedios/8e04644657f2a4fc61f6d400ec6a9881 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pycaret-topic-modeling-the-meat
# initialize the setup
nlp = setup(data = df, target = 'tweet', session_id = 493, custom_stopwords = [ 'rt', 'https', 'http', 'co', 'amp'])
# create the model
lda = create_model('lda', num_topics = 6, multi_core = True)
# label the data using trained model
df_lda = assign_model(lda)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment