lle_manifold

Bases: manifold

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

class lle_manifold(manifold):
    def __init__(self, name: str = 'lle_manifold', *args, **kwargs):
        super().__init__(name=name, *args, **kwargs)

    def init_model(self):
        self.model = LocallyLinearEmbedding(n_neighbors=self.n_neighbors, n_components=self.n_components)