derek-thomas HF staff commited on
Commit
8f38904
1 Parent(s): 1abbec9

Adding topic model

Browse files
Files changed (1) hide show
  1. src/build_nomic.py +2 -1
src/build_nomic.py CHANGED
@@ -115,7 +115,8 @@ def build_nomic(dataset):
115
  # Apply the function
116
  df['subreddit'] = df['content'].apply(extract_subreddit)
117
 
118
- topic_options = NomicTopicOptions(build_topic_model=True, community_description_target_field='subreddit')
 
119
 
120
  delete_old_nomic()
121
 
 
115
  # Apply the function
116
  df['subreddit'] = df['content'].apply(extract_subreddit)
117
 
118
+ topic_options = NomicTopicOptions(build_topic_model=True)
119
+ topic_options.topic_label_field = 'html_content'
120
 
121
  delete_old_nomic()
122