"""
Useful enum definitions realted to classifiers.
"""
from enum import Enum
from lmi_proteins.classifiers.LogisticRegression import LMILogisticRegression
from lmi_proteins.classifiers.GMM import LMIGaussianMixtureModel
from lmi_proteins.classifiers.KMeans import LMIKMeans
from lmi_proteins.classifiers.DummyClassifier import LMIDummyClassifier


class Classifiers(Enum):
    LogReg = LMILogisticRegression
    GMM = LMIGaussianMixtureModel
    KMeans = LMIKMeans
    Dummy = LMIDummyClassifier
