Skip to content

Commit eff34be

Browse files
committed
纵向无损决策树推理报错fix
Signed-off-by: hecy7 <[email protected]>
1 parent 0464061 commit eff34be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fate-serving-federatedml/src/main/java/com/webank/ai/fate/serving/federatedml/model/HeteroSecureBoost.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ public int initModel(byte[] protoMeta, byte[] protoParam) {
7373

7474
protected String getSite(int treeId, int treeNodeId) {
7575
String siteName = this.trees.get(treeId).getTree(treeNodeId).getSitename();
76-
if(siteName!=null&&":".indexOf(siteName)!=0){
76+
if(siteName != null && siteName.contains(":")){
7777
return siteName.split(":")[1];
7878
}else{
7979
return siteName;

0 commit comments

Comments
 (0)