|
@@ -7,6 +7,7 @@
|
|
|
|
|
|
package org.elasticsearch.xpack.core.ml.inference.trainedmodel;
|
|
package org.elasticsearch.xpack.core.ml.inference.trainedmodel;
|
|
|
|
|
|
|
|
+import org.elasticsearch.Version;
|
|
import org.elasticsearch.common.io.stream.StreamInput;
|
|
import org.elasticsearch.common.io.stream.StreamInput;
|
|
import org.elasticsearch.common.io.stream.StreamOutput;
|
|
import org.elasticsearch.common.io.stream.StreamOutput;
|
|
import org.elasticsearch.xcontent.ObjectParser;
|
|
import org.elasticsearch.xcontent.ObjectParser;
|
|
@@ -21,10 +22,10 @@ import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
import java.util.Objects;
|
|
|
|
|
|
|
|
+import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig.CLASSIFICATION_LABELS;
|
|
|
|
+import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig.NUM_TOP_CLASSES;
|
|
|
|
+import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig.RESULTS_FIELD;
|
|
import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig.TOKENIZATION;
|
|
import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig.TOKENIZATION;
|
|
-import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.TextClassificationConfig.CLASSIFICATION_LABELS;
|
|
|
|
-import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.TextClassificationConfig.NUM_TOP_CLASSES;
|
|
|
|
-import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.TextClassificationConfig.RESULTS_FIELD;
|
|
|
|
|
|
|
|
public class TextClassificationConfigUpdate extends NlpConfigUpdate implements NamedXContentObject {
|
|
public class TextClassificationConfigUpdate extends NlpConfigUpdate implements NamedXContentObject {
|
|
|
|
|
|
@@ -96,6 +97,11 @@ public class TextClassificationConfigUpdate extends NlpConfigUpdate implements N
|
|
return NAME;
|
|
return NAME;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+ @Override
|
|
|
|
+ public Version getMinimalSupportedVersion() {
|
|
|
|
+ return Version.V_8_0_0;
|
|
|
|
+ }
|
|
|
|
+
|
|
@Override
|
|
@Override
|
|
public void writeTo(StreamOutput out) throws IOException {
|
|
public void writeTo(StreamOutput out) throws IOException {
|
|
super.writeTo(out);
|
|
super.writeTo(out);
|
|
@@ -237,6 +243,7 @@ public class TextClassificationConfigUpdate extends NlpConfigUpdate implements N
|
|
return this;
|
|
return this;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+ @Override
|
|
public TextClassificationConfigUpdate build() {
|
|
public TextClassificationConfigUpdate build() {
|
|
return new TextClassificationConfigUpdate(
|
|
return new TextClassificationConfigUpdate(
|
|
this.classificationLabels,
|
|
this.classificationLabels,
|