diff --git a/Classification/XGBoost/src/main/java/org/tribuo/classification/xgboost/XGBoostClassificationConverter.java b/Classification/XGBoost/src/main/java/org/tribuo/classification/xgboost/XGBoostClassificationConverter.java index 1abc94786..597c0aa60 100644 --- a/Classification/XGBoost/src/main/java/org/tribuo/classification/xgboost/XGBoostClassificationConverter.java +++ b/Classification/XGBoost/src/main/java/org/tribuo/classification/xgboost/XGBoostClassificationConverter.java @@ -1,5 +1,5 @@ /* - * Copyright (c) 2015-2020, Oracle and/or its affiliates. All rights reserved. + * Copyright (c) 2015, 2022, Oracle and/or its affiliates. All rights reserved. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -16,11 +16,16 @@ package org.tribuo.classification.xgboost; +import com.google.protobuf.Any; +import com.google.protobuf.ByteString; import org.tribuo.Example; import org.tribuo.ImmutableOutputInfo; import org.tribuo.Prediction; import org.tribuo.classification.Label; import org.tribuo.common.xgboost.XGBoostOutputConverter; +import org.tribuo.common.xgboost.protos.XGBoostOutputConverterProto; +import org.tribuo.protos.ProtoSerializableClass; +import org.tribuo.protos.ProtoUtil; import java.util.ArrayList; import java.util.LinkedHashMap; @@ -29,14 +34,36 @@ /** * Converts XGBoost outputs into {@link Label} {@link Prediction}s. */ +@ProtoSerializableClass(version = XGBoostClassificationConverter.CURRENT_VERSION) public final class XGBoostClassificationConverter implements XGBoostOutputConverter