src.wsi#

Submodules#

Package Contents#

Classes#

ClusterChineseWhispers

ClusterCorrelation

WSIModel

Helper class that provides a standard way to create an ABC using

ClusterSpectral

class src.wsi.ClusterChineseWhispers#

Bases: src.wsi.model.WSIModel

n_iters: int#
predict(uses: list[src.use.Use]) → list[int]#
class src.wsi.ClusterCorrelation#

Bases: src.wsi.model.WSIModel

max_senses: int#
max_attempts: int#
max_iters: int#
initial: list[Any]#
split_flag: bool#
predict(uses: list[src.use.Use]) → list[int]#
class src.wsi.WSIModel#

Bases: pydantic.BaseModel, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

wic: WSIModel.wic#
similarity_matrix(use_pairs: list[tuple[src.use.Use, src.use.Use]]) → numpy.typing.NDArray[numpy.float64]#
abstract predict(uses: list[src.use.Use]) → list[int]#
make_freq_dists(clusters: dict[src.use.UseID, int], use_to_grouping: dict[src.use.UseID, str], groupings: tuple[str, str]) → tuple[numpy.typing.NDArray[numpy.int32], numpy.typing.NDArray[numpy.int32]]#
static normalize_cluster(cluster: list[int]) → list[float]#
class src.wsi.ClusterSpectral#

Bases: src.wsi.model.WSIModel

n_clusters: int#
predict(uses: list[src.use.Use]) → list[int]#