nlpatl.models.clustering.sklearn_clustering

class nlpatl.models.clustering.sklearn_clustering.SkLearnClustering(model_name='kmeans', model_config={}, name='sklearn_clustering')[source]

Bases: nlpatl.models.clustering.clustering.Clustering

A wrapper of sci-kit learn clustering class.

Parameters

model_name (str) – sci-kit learn clustering model name. Possible values are kmeans.
model_config (dict) – Model paramateters. Refer to https://scikit-learn.org/stable/modules/classes.html#module-sklearn.cluster
name (str) – Name of this clustering

>>> import nlpatl.models.clustering as nmclu
>>> model = nmclu.SkLearnClustering()

predict_proba(x, predict_config={})[source]

Parameters

x (np.ndarray) – Raw features
predict_config (dict) – Model prediction paramateters. Refer to https://scikit-learn.org/stable/modules/classes.html#module-sklearn.cluster

Returns

Feature and probabilities

Return type

nlptatl.dataset.Dataset

train(x)[source]

Parameters: x (np.ndarray) – Raw features