Skip to content

isomap_manifold

Bases: manifold

Source code in tinybig/koala/manifold/manifold.py
class isomap_manifold(manifold):
    def __init__(self, name: str = 'isomap_manifold', *args, **kwargs):
        super().__init__(name=name, *args, **kwargs)

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