コード例 #1
0
    def fit(self, data):
        """ performs clustering of data

        Parameters
        ----------
        data: np.ndarray
                array of data points to cluster
        merge: int
                minimal number of frames within each cluster. Smaller clusters are merged into next big one
        """
        # if n_clusters is given and no r, estimate n_clusters
        if self.radius is None:
            from htmd.clustering.kcenters import KCenter
            estClust = KCenter(n_clusters=self.n_clusters)
            estClust.fit(data)
            self.radius = estClust.distance.max()
            logger.info("Estimated radius = {}".format(self.radius))

        from pyemma.coordinates.clustering.regspace import RegularSpaceClustering
        self._reg = RegularSpaceClustering(dmin=self.radius)
        self.labels_ = self._reg.fit_transform(data).flatten()
コード例 #2
0
ファイル: test_regspace.py プロジェクト: noinil/PyEMMA
 def setUp(self):
     self.dmin = 0.3
     self.clustering = RegularSpaceClustering(dmin=self.dmin)
     self.src = RandomDataSource()
コード例 #3
0
ファイル: test_regspace.py プロジェクト: ismaelresp/PyEMMA
 def setUp(self):
     self.dmin = 0.3
     self.clustering = RegularSpaceClustering(dmin=self.dmin)
     self.clustering.data_producer = RandomDataSource()