Skip to content

tsne_manifold

Bases: manifold

Source code in tinybig/koala/manifold/manifold.py 
 96
 97
 98
 99
100
101
102
class tsne_manifold(manifold):
    def __init__(self, perplexity: float, name: str = 'tsne_manifold', *args, **kwargs):
        self.perplexity = perplexity
        super().__init__(name=name, *args, **kwargs)

    def init_model(self):
        self.model = TSNE(n_components=self.n_components, perplexity=self.perplexity)