training.py 文件源码

def add_row(self, row_id):
        logger.debug('TreeGaussTrainer.add_row %d', row_id)
        assert row_id not in self._added_rows, row_id
        self._added_rows.add(row_id)

        treegauss_add_row(
            self._data[row_id, :],
            self._tree.tree_grid,
            self._program,
            self._latent[row_id, :, :],
            self._vert_ss,
            self._edge_ss,
            self._feat_ss, )