Skip to content

mds_manifold

Bases: manifold

Source code in tinybig/koala/manifold/manifold.py 
80
81
82
83
84
85
class mds_manifold(manifold):
    def __init__(self, name: str = 'mds_manifold', *args, **kwargs):
        super().__init__(name=name, *args, **kwargs)

    def init_model(self):
        self.model = MDS(n_components=self.n_components)