| import pandas as pd | |
| from turftopic import KeyNMF | |
| def get_topics(sample, text_col, k:int=None): | |
| sample_queries = sample[text_col].tolist() | |
| # Initialize KeyNMF with the desired number of topics | |
| model = KeyNMF(n_components='auto') if k is None else KeyNMF(n_components=k) | |
| out = model.fit_transform(sample_queries) | |
| return pd.concat([sample.reset_index(drop=True), pd.DataFrame(out).reset_index(drop=True)], axis=1), model | |