def config_spec(self) -> types.Spec:
return {
FIELDS_TO_SCRAMBLE_KEY:
types.MultiFieldMatcher(spec='input',
types=['TextSegment'],
select_all=True),
}
def config_spec(self) -> types.Spec:
return {
# Requires a substitution string. Include a default.
SUBSTITUTIONS_KEY:
types.TextSegment(default='great -> terrible'),
FIELDS_TO_REPLACE_KEY:
types.MultiFieldMatcher(spec='input',
types=['TextSegment'],
select_all=True),
}
def config_spec(self) -> types.Spec:
return {
NUM_EXAMPLES_KEY:
types.TextSegment(default=str(NUM_EXAMPLES_DEFAULT)),
MAX_ABLATIONS_KEY:
types.TextSegment(default=str(MAX_ABLATIONS_DEFAULT)),
PREDICTION_KEY:
types.FieldMatcher(spec="output",
types=["MulticlassPreds", "RegressionScore"]),
REGRESSION_THRESH_KEY:
types.TextSegment(default=str(REGRESSION_THRESH_DEFAULT)),
FIELDS_TO_ABLATE_KEY:
types.MultiFieldMatcher(spec="input",
types=["TextSegment", "SparseMultilabel"],
select_all=True),
}
def config_spec(self) -> types.Spec:
return {
NUM_EXAMPLES_KEY:
types.TextSegment(default=str(NUM_EXAMPLES_DEFAULT)),
MAX_FLIPS_KEY:
types.TextSegment(default=str(MAX_FLIPS_DEFAULT)),
TOKENS_TO_IGNORE_KEY:
types.Tokens(default=TOKENS_TO_IGNORE_DEFAULT),
PREDICTION_KEY:
types.FieldMatcher(spec="output",
types=["MulticlassPreds", "RegressionScore"]),
REGRESSION_THRESH_KEY:
types.TextSegment(default=str(REGRESSION_THRESH_DEFAULT)),
FIELDS_TO_HOTFLIP_KEY:
types.MultiFieldMatcher(spec="input",
types=["Tokens"],
select_all=True),
}