marvinscham/masterthesis-playground
1
0
Fork 0
mirror of https://github.com/marvinscham/masterthesis-playground.git synced 2026-03-22 08:22:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

BERTopic cleanup

Browse Source
This commit is contained in:
Marvin Scham
2026-02-08 22:43:53 +01:00
parent b2da597b18
commit c98a1d0c6e
8 changed files with 1400 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bertopic/bertopic_autotune_sorter.py
View File

@@ -2,12 +2,12 @@ import json
import matplotlib.pyplot as plt
with open("history.json", "r") as f:
with open("output/autotune.json", "r") as f:
history = json.load(f)
history = sorted(history, key=lambda x: x["metrics"]["combined_score"], reverse=True)
history = sorted(history, key=lambda x: x["metrics"]["combined_score"], reverse=False)
with open("history_sorted.json", "w") as f:
with open("output/autotune_sorted.json", "w") as f:
json.dump(history, f, indent=2)