neural_compressor.metric

Intel Neural Compressor Metric.

Submodules

• neural_compressor.metric.bleu
• neural_compressor.metric.bleu_util
• neural_compressor.metric.coco_label_map
• neural_compressor.metric.coco_tools
• neural_compressor.metric.evaluate_squad
• neural_compressor.metric.f1
• neural_compressor.metric.metric

Package Contents

Classes

METRICS	Intel Neural Compressor Metrics.
Metric	A wrapper of the information needed to construct a Metric.
BaseMetric	The base class of Metric.
TensorflowTopK	Compute Top-k Accuracy classification score for Tensorflow model.
COCOmAPv2	Compute mean average precision of the detection task.
SquadF1	Evaluate for v1.1 of the SQuAD dataset.
GeneralTopK	Compute Top-k Accuracy classification score.

Functions

metric_registry(metric_type, framework)	Decorate for registering all Metric subclasses.
register_customer_metric(user_metric, framework)	Register customer metric class or a dict of built-in metric configures.