Example #1
0
    def prepare_data(self, guest_data):
        LOGGER.info("@ start guest prepare_data")
        guest_features_dict, guest_label_dict, guest_sample_indexes = convert_instance_table_to_dict(
            guest_data)
        guest_sample_indexes = np.array(guest_sample_indexes)
        LOGGER.debug("@ send guest_sample_indexes shape" +
                     str(guest_sample_indexes.shape))
        self._do_remote(guest_sample_indexes,
                        name=self.transfer_variable.guest_sample_indexes.name,
                        tag=self.transfer_variable.generate_transferid(
                            self.transfer_variable.guest_sample_indexes),
                        role=consts.HOST,
                        idx=-1)
        host_sample_indexes = self._do_get(
            name=self.transfer_variable.host_sample_indexes.name,
            tag=self.transfer_variable.generate_transferid(
                self.transfer_variable.host_sample_indexes),
            idx=-1)[0]

        LOGGER.debug("@ receive host_sample_indexes len" +
                     str(len(host_sample_indexes)))
        guest_features, overlap_indexes, non_overlap_indexes, guest_label = overlapping_samples_converter(
            guest_features_dict, guest_sample_indexes, host_sample_indexes,
            guest_label_dict)
        return guest_features, overlap_indexes, non_overlap_indexes, guest_label
Example #2
0
    def prepare_data(self, host_data):
        LOGGER.info("@ start host prepare data")
        host_features_dict, _, host_sample_indexes = convert_instance_table_to_dict(
            host_data)
        host_sample_indexes = np.array(host_sample_indexes)

        self._do_remote(host_sample_indexes,
                        name=self.transfer_variable.host_sample_indexes.name,
                        tag=self.transfer_variable.generate_transferid(
                            self.transfer_variable.host_sample_indexes),
                        role=consts.GUEST,
                        idx=-1)

        guest_sample_indexes = self._do_get(
            name=self.transfer_variable.guest_sample_indexes.name,
            tag=self.transfer_variable.generate_transferid(
                self.transfer_variable.guest_sample_indexes),
            idx=-1)[0]

        host_features, overlap_indexes, _ = overlapping_samples_converter(
            host_features_dict, host_sample_indexes, guest_sample_indexes)
        return host_features, overlap_indexes