Skip to content

Commit 97e0ce9

Browse files
Added test case for dynamic KeyNMF
1 parent c7b835c commit 97e0ce9

1 file changed

Lines changed: 5 additions & 9 deletions

File tree

tests/test_integration.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from datetime import datetime
21
import tempfile
2+
from datetime import datetime
33
from pathlib import Path
44

55
import numpy as np
@@ -8,13 +8,8 @@
88
from sentence_transformers import SentenceTransformer
99
from sklearn.datasets import fetch_20newsgroups
1010

11-
from turftopic import (
12-
GMM,
13-
AutoEncodingTopicModel,
14-
ClusteringTopicModel,
15-
KeyNMF,
16-
SemanticSignalSeparation,
17-
)
11+
from turftopic import (GMM, AutoEncodingTopicModel, ClusteringTopicModel,
12+
KeyNMF, SemanticSignalSeparation)
1813

1914

2015
def generate_dates(
@@ -75,8 +70,9 @@ def generate_dates(
7570
n_reduce_to=5,
7671
feature_importance="soft-c-tf-idf",
7772
encoder=trf,
78-
reduction_method="smallest"
73+
reduction_method="smallest",
7974
),
75+
KeyNMF(5, encoder=trf),
8076
]
8177

8278

0 commit comments

Comments
 (0)