tsne_manifold

Bases: manifold

Manifold learning using the t-SNE algorithm.

This class implements dimensionality reduction using the t-SNE algorithm, which is effective for visualizing high-dimensional data.

Source code in tinybig/koala/manifold/manifold.py

class tsne_manifold(manifold):
    """
    Manifold learning using the t-SNE algorithm.

    This class implements dimensionality reduction using the t-SNE algorithm, which is effective for visualizing
    high-dimensional data.
    """
    def __init__(self, perplexity: float = 5.0, name: str = 'tsne_manifold', *args, **kwargs):
        """
        Initialize the t-SNE manifold learning class.

        Parameters
        ----------
        perplexity : float
            The perplexity parameter for t-SNE.
        name : str, optional
            The name of the manifold learning method. Default is 'tsne_manifold'.
        *args, **kwargs
            Additional arguments for the base class.
        """
        self.perplexity = perplexity
        super().__init__(name=name, *args, **kwargs)

    def init_model(self):
        """
        Initialize the t-SNE model.
        """
        self.model = TSNE(n_components=self.n_components, perplexity=self.perplexity)

__init__(perplexity=5.0, name='tsne_manifold', *args, **kwargs)

Initialize the t-SNE manifold learning class.

Parameters:

Name	Type	Description	Default
perplexity	float	The perplexity parameter for t-SNE.	5.0
name	str	The name of the manifold learning method. Default is 'tsne_manifold'.	'tsne_manifold'
*args		Additional arguments for the base class.	()
**kwargs		Additional arguments for the base class.	()

Source code in tinybig/koala/manifold/manifold.py

def __init__(self, perplexity: float = 5.0, name: str = 'tsne_manifold', *args, **kwargs):
    """
    Initialize the t-SNE manifold learning class.

    Parameters
    ----------
    perplexity : float
        The perplexity parameter for t-SNE.
    name : str, optional
        The name of the manifold learning method. Default is 'tsne_manifold'.
    *args, **kwargs
        Additional arguments for the base class.
    """
    self.perplexity = perplexity
    super().__init__(name=name, *args, **kwargs)

init_model()

Initialize the t-SNE model.

Source code in tinybig/koala/manifold/manifold.py

def init_model(self):
    """
    Initialize the t-SNE model.
    """
    self.model = TSNE(n_components=self.n_components, perplexity=self.perplexity)