Source code for flyqma.annotation.model_selection.bivariate

from ..classification import BivariateMixtureClassifier
from .univariate import UnivariateModelSelection


class BivariateModelSelection(UnivariateModelSelection):
    """ Bivariate extension for model selection. """

    @staticmethod
    def load_model(path):
        """ Load model from <path> directory. """
        return BivariateMixtureClassifier.load(path)

    @staticmethod
    def fit_model(values, num_components, num_labels, **kwargs):
        """ Fit model with specified number of components. """
        return BivariateMixtureClassifier(values,
                                        num_components=num_components,
                                        num_labels=num_labels,
                                        **kwargs)