Module org.elasticsearch.xcore
Class ZeroShotClassificationConfig
java.lang.Object
org.elasticsearch.xpack.core.ml.inference.trainedmodel.ZeroShotClassificationConfig
- All Implemented Interfaces:
NamedWriteable,VersionedNamedWriteable,Writeable,org.elasticsearch.xcontent.ToXContent,org.elasticsearch.xcontent.ToXContentObject,InferenceConfig,LenientlyParsedInferenceConfig,NlpConfig,StrictlyParsedInferenceConfig,NamedXContentObject
This builds out a 0-shot classification task.
The 0-shot methodology assumed is MNLI optimized task. For further info see: https://arxiv.org/abs/1909.00161
-
Nested Class Summary
Nested classes/interfaces inherited from interface org.elasticsearch.xcontent.ToXContent
org.elasticsearch.xcontent.ToXContent.DelegatingMapParams, org.elasticsearch.xcontent.ToXContent.MapParams, org.elasticsearch.xcontent.ToXContent.ParamsNested classes/interfaces inherited from interface org.elasticsearch.common.io.stream.Writeable
Writeable.Reader<V>, Writeable.Writer<V> -
Field Summary
FieldsModifier and TypeFieldDescriptionstatic final org.elasticsearch.xcontent.ParseFieldstatic final org.elasticsearch.xcontent.ParseFieldstatic final org.elasticsearch.xcontent.ParseFieldstatic final StringFields inherited from interface org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceConfig
DEFAULT_RESULTS_FIELD, DEFAULT_TOP_CLASSES_RESULTS_FIELD, RESULTS_FIELDFields inherited from interface org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig
CLASSIFICATION_LABELS, MINIMUM_NLP_SUPPORTED_VERSION, NUM_TOP_CLASSES, TOKENIZATION, VOCABULARYFields inherited from interface org.elasticsearch.xcontent.ToXContent
EMPTY, EMPTY_PARAMS -
Constructor Summary
ConstructorsConstructorDescriptionZeroShotClassificationConfig(List<String> classificationLabels, VocabularyConfig vocabularyConfig, Tokenization tokenization, String hypothesisTemplate, Boolean isMultiLabel, List<String> labels, String resultsField) -
Method Summary
Modifier and TypeMethodDescriptionapply(InferenceConfigUpdate update) Return a copy of this with the settings updated by the values inupdate.booleanstatic ZeroShotClassificationConfigfromXContentLenient(org.elasticsearch.xcontent.XContentParser parser) static ZeroShotClassificationConfigfromXContentStrict(org.elasticsearch.xcontent.XContentParser parser) All nodes in the cluster must have at least this MlConfigVersion attributeAll communication in the cluster must use at least this versiongetName()inthashCode()booleanbooleanbooleanisTargetTypeSupported(TargetType targetType) org.elasticsearch.xcontent.XContentBuildertoXContent(org.elasticsearch.xcontent.XContentBuilder builder, org.elasticsearch.xcontent.ToXContent.Params params) voidwriteTo(StreamOutput out) Methods inherited from class java.lang.Object
clone, finalize, getClass, notify, notifyAll, toString, wait, wait, waitMethods inherited from interface org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceConfig
getDefaultInput, getMinimalSupportedVersion, getMinLicenseSupported, getMinLicenseSupportedForAction, incompatibleUpdateException, requestingImportance, validateTrainedModelInputMethods inherited from interface org.elasticsearch.xpack.core.ml.inference.trainedmodel.NlpConfig
supportsIngestPipeline, supportsPipelineAggregation, supportsSearchRescorerMethods inherited from interface org.elasticsearch.xcontent.ToXContentObject
isFragmentMethods inherited from interface org.elasticsearch.common.io.stream.VersionedNamedWriteable
supportsVersion
-
Field Details
-
NAME
- See Also:
-
HYPOTHESIS_TEMPLATE
public static final org.elasticsearch.xcontent.ParseField HYPOTHESIS_TEMPLATE -
MULTI_LABEL
public static final org.elasticsearch.xcontent.ParseField MULTI_LABEL -
LABELS
public static final org.elasticsearch.xcontent.ParseField LABELS
-
-
Constructor Details
-
ZeroShotClassificationConfig
public ZeroShotClassificationConfig(List<String> classificationLabels, @Nullable VocabularyConfig vocabularyConfig, @Nullable Tokenization tokenization, @Nullable String hypothesisTemplate, @Nullable Boolean isMultiLabel, @Nullable List<String> labels, @Nullable String resultsField) -
ZeroShotClassificationConfig
- Throws:
IOException
-
-
Method Details
-
fromXContentStrict
public static ZeroShotClassificationConfig fromXContentStrict(org.elasticsearch.xcontent.XContentParser parser) -
fromXContentLenient
public static ZeroShotClassificationConfig fromXContentLenient(org.elasticsearch.xcontent.XContentParser parser) -
writeTo
- Specified by:
writeToin interfaceWriteable- Throws:
IOException
-
toXContent
public org.elasticsearch.xcontent.XContentBuilder toXContent(org.elasticsearch.xcontent.XContentBuilder builder, org.elasticsearch.xcontent.ToXContent.Params params) throws IOException - Specified by:
toXContentin interfaceorg.elasticsearch.xcontent.ToXContent- Throws:
IOException
-
getWriteableName
- Specified by:
getWriteableNamein interfaceNamedWriteable- Specified by:
getWriteableNamein interfaceVersionedNamedWriteable
-
isTargetTypeSupported
- Specified by:
isTargetTypeSupportedin interfaceInferenceConfig
-
apply
Description copied from interface:InferenceConfigReturn a copy of this with the settings updated by the values inupdate.- Specified by:
applyin interfaceInferenceConfig- Parameters:
update- The update to apply- Returns:
- A new updated config
-
getMinimalSupportedMlConfigVersion
Description copied from interface:InferenceConfigAll nodes in the cluster must have at least this MlConfigVersion attribute- Specified by:
getMinimalSupportedMlConfigVersionin interfaceInferenceConfig
-
getMinimalSupportedTransportVersion
Description copied from interface:InferenceConfigAll communication in the cluster must use at least this version- Specified by:
getMinimalSupportedTransportVersionin interfaceInferenceConfig
-
getName
- Specified by:
getNamein interfaceNamedXContentObject- Returns:
- The name of the XContentObject that is to be serialized
-
equals
-
hashCode
public int hashCode() -
getVocabularyConfig
- Specified by:
getVocabularyConfigin interfaceNlpConfig- Returns:
- the vocabulary configuration that allows retrieving it
-
getTokenization
- Specified by:
getTokenizationin interfaceNlpConfig- Returns:
- the model tokenization parameters
-
getClassificationLabels
-
isMultiLabel
public boolean isMultiLabel() -
getHypothesisTemplate
-
getLabels
-
getResultsField
- Specified by:
getResultsFieldin interfaceInferenceConfig
-
isAllocateOnly
public boolean isAllocateOnly()- Specified by:
isAllocateOnlyin interfaceInferenceConfig
-