kisejin commited on
Commit
b105c38
·
verified ·
1 Parent(s): 8f560ee

Update BERTopic/my_topic_modeling.py

Browse files
Files changed (1) hide show
  1. BERTopic/my_topic_modeling.py +1 -1
BERTopic/my_topic_modeling.py CHANGED
@@ -595,7 +595,7 @@ def working(args: argparse.Namespace, name_dataset: str):
595
 
596
  if __name__ == "__main__":
597
  args = init_args()
598
- if args.train_both == 'yes:
599
  working(args, 'booking')
600
  working(args, 'tripadvisor')
601
  else:
 
595
 
596
  if __name__ == "__main__":
597
  args = init_args()
598
+ if args.train_both == 'yes':
599
  working(args, 'booking')
600
  working(args, 'tripadvisor')
601
  else: