File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
tensorflow_addons/metrics Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 1515#### Standard API
1616In order to conform with the current API standard, all metrics
1717must:
18- * Be a ` tf.function ` .
19- * Have the signature ` fn(input, axis=-1, name=None) ` .
18+ * Inherit from ` tf.metrics.Metric ` .
2019 * [ Register as a keras global object] ( https://github.com/tensorflow/addons/blob/master/tensorflow_addons/utils/python/keras_utils.py )
2120 so it can be serialized properly.
2221 * Add the addon to the ` py_library ` in this sub-package's BUILD file.
2322
2423#### Testing Requirements
25- * Simple unittests that demonstrate the layer is behaving as expected.
24+ * Simple unittests that demonstrate the metric is behaving as expected.
2625 * When applicable, run all unittests with TensorFlow's
2726 ` @run_in_graph_and_eager_modes ` (for test method)
2827 or ` run_all_in_graph_and_eager_modes ` (for TestCase subclass)
You can’t perform that action at this time.
0 commit comments