(self)
| 265 | return all_results |
| 266 | |
| 267 | def config_spec(self) -> types.Spec: |
| 268 | return { |
| 269 | TARGET_INFO_KEY: types.SalienceTargetInfo(), |
| 270 | # TODO(b/205996131): remove TARGET_HEAD_KEY field |
| 271 | TARGET_HEAD_KEY: types.SingleFieldMatcher( |
| 272 | spec='output', |
| 273 | types=[c.__name__ for c in _SUPPORTED_PRED_TYPES], |
| 274 | required=False, |
| 275 | ), |
| 276 | # TODO(b/205996131): remove CLASS_KEY field |
| 277 | CLASS_KEY: types.TextSegment(default='-1', required=False), |
| 278 | MASK_KEY: types.TextSegment(default=self._mask_token, required=False), |
| 279 | KERNEL_WIDTH_KEY: types.TextSegment( |
| 280 | default=self._kernel_width, required=False |
| 281 | ), |
| 282 | NUM_SAMPLES_KEY: types.TextSegment( |
| 283 | default=self._num_samples, required=False |
| 284 | ), |
| 285 | SEED_KEY: types.TextSegment(default=self._seed, required=False), |
| 286 | } |
| 287 | |
| 288 | def is_compatible(self, model: lit_model.Model, |
| 289 | dataset: lit_dataset.Dataset) -> bool: |
no outgoing calls