diff --git a/AllMeans/all_means.py b/AllMeans/all_means.py index 8e70a7a..9d1c95b 100644 --- a/AllMeans/all_means.py +++ b/AllMeans/all_means.py @@ -166,7 +166,6 @@ def model_topics(self, early_stop = 2, verbose = False, model = "distilbert-base if verbose: print("-" * 50) print(f"Early stop triggered. Logged scores: {[round(scr, 4) for scr in self.avg_scores]}") # rounded for readability - # print("-" * 50) break order_centroids = kmeans.cluster_centers_.argsort()[:, ::-1] @@ -187,7 +186,6 @@ def model_topics(self, early_stop = 2, verbose = False, model = "distilbert-base if best_selection: if verbose: print(f"Best selection before assigning clusters: {best_selection}") - # print("-" * 50) self._assign_clusters_to_sentences(clusters, best_selection, verbose) return self.clusters diff --git a/CHANGELOG.md b/CHANGELOG.md new file mode 100644 index 0000000..b2f6d3f --- /dev/null +++ b/CHANGELOG.md @@ -0,0 +1,13 @@ +# Changelog + +All notable changes to this AllMeans will be documented in this file. + +## [Future Additions] + +- AllMeans.model_topics() to also take arg 'exclusions' (name unconfirmed) that will allow the user to pass a list of strings to be excluded from considered topics. + +## [Unreleased] + +### Added + +- Initial features and documentation. AllMeans.model_topics() is currently the main method, with args 'early_stop', 'verbose', and 'model'. \ No newline at end of file