search

FLAIR-THU / VFLAIR

THU-AIR Vertical Federated Learning general, extensible and light-weight framework
68 stars 18 forks source link

Can not find get_num_parties_per_process method in XGBoostNode #3

Closed yooopan closed 10 months ago

yooopan commented 11 months ago

Can not find get_num_parties_per_process method in XGBoostNode 

        if self.use_only_active_party:
            self.find_split_per_party(
                self.active_party_id, 1, sum_grad, sum_hess, tot_cnt
            )
        else:
            if self.n_job == 1:
                self.find_split_per_party(
                    0, self.num_parties, sum_grad, sum_hess, tot_cnt
                )
            else:
                num_parties_per_thread = self.get_num_parties_per_process(
                    self.n_job, self.num_parties
                )