pysummarization.clusterabledoc package¶
Submodules¶
pysummarization.clusterabledoc.k_means module¶
-
class
pysummarization.clusterabledoc.k_means.
KMeans
(cluster_num=2, max_iter=100, init_noise_arr=None)[source]¶ Bases:
pysummarization.clusterable_doc.ClusterableDoc
K-Means method.
-
centroid_arr
¶ getter
-
inference
(document_arr)[source]¶ Inferencing.
Parameters: document_arr – np.ndarray of sentences vectors. - Retruns:
- np.ndarray of labeled data.
-