tommy.support.model_type

Classes

class tommy.support.model_type.ModelType(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]

Bases: Enum

An enumeration of the types of models.

BERTopic = 3

LDA = 1

NMF = 2

static from_string(model_type: str) → ModelType[source]: Convert a string to a ModelType. :param model_type: The string to convert :return: The ModelType

static to_string(model_type: ModelType) → str[source]: Convert a ModelType to a string. :param model_type: The ModelType to convert :return: The string