update

Koukyosyumei · Aug 18, 2023 · f19bcbf · f19bcbf
1 parent e67aa42
commit f19bcbf
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/src/aijack/collaborative/tree/a.out b/src/aijack/collaborative/tree/a.out
diff --git a/src/aijack/collaborative/tree/core/nodeapi.h b/src/aijack/collaborative/tree/core/nodeapi.h
@@ -171,9 +171,8 @@ template <typename NodeType> struct NodeAPI {
       if (temp_node->is_leaf()) {
         return temp_node->val;
       } else {
-        if (true) {
-          //(node->parties[temp_node->party_id].is_left(temp_node->record_id,
-          //                                            xi)) {
+        if (node->parties[temp_node->party_id].is_left(temp_node->record_id,
+                                                       xi)) {
           que.push(temp_node->left);
         } else {
           que.push(temp_node->right);

diff --git a/src/aijack/collaborative/tree/secureboost/node.h b/src/aijack/collaborative/tree/secureboost/node.h
@@ -65,6 +65,7 @@ struct SecureBoostNode : Node<SecureBoostParty> {
 
   SecureBoostNode &operator=(const SecureBoostNode &other) {
     if (this != &other) {
+      parties = other.parties;
       y = other.y;
       idxs = other.idxs;
       val = other.val;

diff --git a/src/aijack/collaborative/tree/xgboost/node.h b/src/aijack/collaborative/tree/xgboost/node.h
@@ -94,6 +94,7 @@ struct XGBoostNode : public Node<XGBoostParty> {
 
   XGBoostNode &operator=(const XGBoostNode &other) {
     if (this != &other) {
+      parties = other.parties;
       y = other.y;
       idxs = other.idxs;
       val = other.val;