training.py 文件源码

def set_edges(self, edges):
        TreeTrainer.set_edges(self, edges)
        V, E, K, M = self._VEKM
        assignments = self._assignments[sorted(self._added_rows), :]
        for e, v1, v2 in self._tree.tree_grid.T:
            self._edge_ss[e, :, :] = count_pairs(assignments, v1, v2, M)
        np.add(self._edge_ss, self._edge_prior, out=self._edge_probs)