Creates an extractor for extracting example weights.
tfma.extractors.ExampleWeightsExtractor(
eval_config: tfma.EvalConfig
) -> tfma.extractors.Extractor
The extractor's PTransform uses the config's ModelSpec.example_weight_key(s)
to lookup the associated example weight values stored as features under the
tfma.FEATURES_KEY (and optionally tfma.TRANSFORMED_FEATURES_KEY) in extracts.
The resulting values are then added to the extracts under the key
tfma.EXAMPLE_WEIGHTS_KEY.
Args |
|---|
eval_config
|
Eval config.
|
Returns |
|---|
|
Extractor for extracting example weights.
|
View source on GitHub