diff --git a/data-computer/v1/lda.py b/data-computer/v1/lda.py index 478d62e..1611e05 100755 --- a/data-computer/v1/lda.py +++ b/data-computer/v1/lda.py @@ -123,14 +123,17 @@ words_weights = [] for word, word_weight in lda_model.show_topic(topic_id): words.append(word) - words_weights.append(words_weights) + words_weights.append(str(word_weight)) topic_info[f"topic_{topic_id + 1}"]["words"] = words topic_info[f"topic_{topic_id + 1}"]["words_weights"] = words_weights topic_info[f"topic_{topic_id + 1}"]["topic_weight"] = str(topic_weight) line["value"]={} line["value"]["topics"]=topic_info - line["value"]["best_topic"]=max_topic(topic_info) + try: + line["value"]["best_topic"]=max_topic(topic_info) + except: + line["value"]["best_topic"]="n/a" sys.stdout.write(json.dumps(line)) sys.stdout.write("\n")