Revert "doctest addons (#677)" (#2068)

This reverts commit 706e22f.

Author: WindQAQ

tensorflow_addons/conftest.py

@@ -14,20 +14,6 @@
     pytest_collection_modifyitems,
 )
 
-import numpy as np
-import pytest
-
-import tensorflow as tf
-import tensorflow_addons as tfa
-
-
 # fixtures present in this file will be available
 # when running tests and can be referenced with strings
 # https://docs.pytest.org/en/latest/fixture.html#conftest-py-sharing-fixture-functions
-
-
-@pytest.fixture(autouse=True)
-def add_np(doctest_namespace):
-    doctest_namespace["np"] = np
-    doctest_namespace["tf"] = tf
-    doctest_namespace["tfa"] = tfa

tensorflow_addons/layers/wrappers.py

@@ -16,7 +16,6 @@
 import logging
 
 import tensorflow as tf
-
 from typeguard import typechecked
 
 
@@ -31,25 +30,21 @@ class WeightNormalization(tf.keras.layers.Wrapper):
     Training of Deep Neural Networks: https://arxiv.org/abs/1602.07868
     Tim Salimans, Diederik P. Kingma (2016)
     WeightNormalization wrapper works for keras and tf layers.
-
-    Usage:
-
-    >>> net = tfa.layers.WeightNormalization(
-    ...   tf.keras.layers.Conv2D(2, 2, activation='relu'),
-    ...   input_shape=(32, 32, 3),
-    ...   data_init=True)(np.random.rand(32, 32, 3, 1).astype('f'))
-    >>> net = tfa.layers.WeightNormalization(
-    ...   tf.keras.layers.Conv2D(16, 2, activation='relu'),
-    ...   data_init=True)(net)
-    >>> net = tfa.layers.WeightNormalization(
-    ...   tf.keras.layers.Dense(120, activation='relu'),
-    ...   data_init=True)(net)
-    >>> net = tfa.layers.WeightNormalization(
-    ...   tf.keras.layers.Dense(2),
-    ...   data_init=True)(net)
-    >>> net.shape
-    TensorShape([32, 30, 1, 2])
-
+    ```python
+      net = WeightNormalization(
+          tf.keras.layers.Conv2D(2, 2, activation='relu'),
+          input_shape=(32, 32, 3),
+          data_init=True)(x)
+      net = WeightNormalization(
+          tf.keras.layers.Conv2D(16, 5, activation='relu'),
+          data_init=True)(net)
+      net = WeightNormalization(
+          tf.keras.layers.Dense(120, activation='relu'),
+          data_init=True)(net)
+      net = WeightNormalization(
+          tf.keras.layers.Dense(n_classes),
+          data_init=True)(net)
+    ```
     Arguments:
       layer: a layer instance.
       data_init: If `True` use data dependent variable initialization

tensorflow_addons/losses/focal_loss.py

@@ -15,7 +15,6 @@
 """Implements Focal loss."""
 
 import tensorflow as tf
-
 import tensorflow.keras.backend as K
 
 from tensorflow_addons.utils.keras_utils import LossFunctionWrapper
@@ -38,11 +37,15 @@ class SigmoidFocalCrossEntropy(LossFunctionWrapper):
 
     Usage:
 
-    >>> fl = tfa.losses.SigmoidFocalCrossEntropy()
-    >>> loss = fl([[0.97], [0.91], [0.03]], [[1.0], [1.0], [0.0]])
-    >>> loss
-    <tf.Tensor: shape=(3,), dtype=float32, numpy=array([0.00010971, 0.00329749, 0.00030611], dtype=float32)>
-
+    ```python
+    fl = tfa.losses.SigmoidFocalCrossEntropy()
+    loss = fl(
+      y_true = [[1.0], [1.0], [0.0]],
+      y_pred = [[0.97], [0.91], [0.03]])
+    print('Loss: ', loss.numpy())  # Loss: [6.8532745e-06,
+                                            1.9097870e-04,
+                                            2.0559824e-05]
+    ```
     Usage with tf.keras API:
 
     ```python

tensorflow_addons/losses/giou_loss.py

@@ -33,13 +33,13 @@ class GIoULoss(LossFunctionWrapper):
 
     Usage:
 
-    >>> gl = tfa.losses.GIoULoss()
-    >>> boxes1 = tf.constant([[4.0, 3.0, 7.0, 5.0], [5.0, 6.0, 10.0, 7.0]])
-    >>> boxes2 = tf.constant([[3.0, 4.0, 6.0, 8.0], [14.0, 14.0, 15.0, 15.0]])
-    >>> loss = gl(boxes1, boxes2)
-    >>> loss
-    <tf.Tensor: shape=(), dtype=float32, numpy=1.5041667>
-
+    ```python
+    gl = tfa.losses.GIoULoss()
+    boxes1 = tf.constant([[4.0, 3.0, 7.0, 5.0], [5.0, 6.0, 10.0, 7.0]])
+    boxes2 = tf.constant([[3.0, 4.0, 6.0, 8.0], [14.0, 14.0, 15.0, 15.0]])
+    loss = gl(boxes1, boxes2)
+    print('Loss: ', loss.numpy())  # Loss: [1.07500000298023224, 1.9333333373069763]
+    ```
     Usage with tf.keras API:
 
     ```python

tensorflow_addons/losses/quantiles.py

@@ -84,11 +84,15 @@ class PinballLoss(LossFunctionWrapper):
     See: https://en.wikipedia.org/wiki/Quantile_regression
 
     Usage:
+    ```python
+    pinball = tfa.losses.PinballLoss(tau=.1)
+    loss = pinball([0., 0., 1., 1.], [1., 1., 1., 0.])
+
+    # loss = max(0.1 * (y_true - y_pred), (0.1 - 1) * (y_true - y_pred))
+    #      = (0.9 + 0.9 + 0 + 0.1) / 4
 
-    >>> pinball = tfa.losses.PinballLoss(tau=.1)
-    >>> loss = pinball([0., 0., 1., 1.], [1., 1., 1., 0.])
-    >>> loss
-    <tf.Tensor: shape=(), dtype=float32, numpy=0.475>
+    print('Loss: ', loss.numpy())  # Loss: 0.475
+    ```
 
     Usage with the `compile` API:
 

tensorflow_addons/metrics/cohens_kappa.py

@@ -15,7 +15,6 @@
 """Implements Cohen's Kappa."""
 
 import tensorflow as tf
-
 import numpy as np
 import tensorflow.keras.backend as K
 from tensorflow.keras.metrics import Metric
@@ -28,43 +27,39 @@
 @tf.keras.utils.register_keras_serializable(package="Addons")
 class CohenKappa(Metric):
     """Computes Kappa score between two raters.
+
     The score lies in the range [-1, 1]. A score of -1 represents
     complete disagreement between two raters whereas a score of 1
     represents complete agreement between the two raters.
     A score of 0 means agreement by chance.
+
     Note: As of now, this implementation considers all labels
     while calculating the Cohen's Kappa score.
 
     Usage:
 
-    >>> actuals = np.array([4, 4, 3, 4, 2, 4, 1, 1], dtype=np.int32)
-    >>> preds = np.array([4, 4, 3, 4, 4, 2, 1, 1], dtype=np.int32)
-    >>> weights = np.array([1, 1, 2, 5, 10, 2, 3, 3], dtype=np.int32)
-    >>> m = tfa.metrics.CohenKappa(num_classes=5, sparse_labels=True)
-    >>> m.update_state(actuals, preds)
-    <tf.Tensor: shape=(5, 5), dtype=float32, numpy=
-    array([[0., 0., 0., 0., 0.],
-           [0., 2., 0., 0., 0.],
-           [0., 0., 0., 0., 1.],
-           [0., 0., 0., 1., 0.],
-           [0., 0., 1., 0., 3.]], dtype=float32)>
-    >>> m.result().numpy()
-    0.61904764
-    >>> m = tfa.metrics.CohenKappa(num_classes=5, sparse_labels=True)
-    >>> m.update_state(actuals, preds, sample_weight=weights)
-    <tf.Tensor: shape=(5, 5), dtype=float32, numpy=
-    array([[ 0.,  0.,  0.,  0.,  0.],
-           [ 0.,  6.,  0.,  0.,  0.],
-           [ 0.,  0.,  0.,  0., 10.],
-           [ 0.,  0.,  0.,  2.,  0.],
-           [ 0.,  0.,  2.,  0.,  7.]], dtype=float32)>
-    >>> m.result().numpy()
-    0.37209308
+    ```python
+    actuals = np.array([4, 4, 3, 4, 2, 4, 1, 1], dtype=np.int32)
+    preds = np.array([4, 4, 3, 4, 4, 2, 1, 1], dtype=np.int32)
+    weights = np.array([1, 1, 2, 5, 10, 2, 3, 3], dtype=np.int32)
+
+    m = tfa.metrics.CohenKappa(num_classes=5, sparse_labels=True)
+    m.update_state(actuals, preds)
+    print('Final result: ', m.result().numpy()) # Result: 0.61904764
+
+    # To use this with weights, sample_weight argument can be used.
+    m = tfa.metrics.CohenKappa(num_classes=5, sparse_labels=True)
+    m.update_state(actuals, preds, sample_weight=weights)
+    print('Final result: ', m.result().numpy()) # Result: 0.37209308
+    ```
 
     Usage with tf.keras API:
+
+    ```python
     model = tf.keras.models.Model(inputs, outputs)
     model.add_metric(tfa.metrics.CohenKappa(num_classes=5)(outputs))
     model.compile('sgd', loss='mse')
+    ```
     """
 
     @typechecked

tensorflow_addons/metrics/matthews_correlation_coefficient.py

@@ -42,13 +42,18 @@ class MatthewsCorrelationCoefficient(tf.keras.metrics.Metric):
           ((TP + FP) * (TP + FN) * (TN + FP ) * (TN + FN))^(1/2)
 
     Usage:
-
-    >>> actuals = tf.constant([[1.0], [1.0], [1.0], [0.0]], dtype=tf.float32)
-    >>> preds = tf.constant([[1.0], [0.0], [1.0], [1.0]], dtype=tf.float32)
-    >>> mcc = tfa.metrics.MatthewsCorrelationCoefficient(num_classes=1)
-    >>> mcc.update_state(actuals, preds)
-    >>> mcc.result()
-    <tf.Tensor: shape=(1,), dtype=float32, numpy=array([-0.33333334], dtype=float32)>
+    ```python
+    actuals = tf.constant([[1.0], [1.0], [1.0], [0.0]],
+             dtype=tf.float32)
+    preds = tf.constant([[1.0], [0.0], [1.0], [1.0]],
+             dtype=tf.float32)
+    # Matthews correlation coefficient
+    mcc = MatthewsCorrelationCoefficient(num_classes=1)
+    mcc.update_state(actuals, preds)
+    print('Matthews correlation coefficient is:',
+    mcc.result().numpy())
+    # Matthews correlation coefficient is : -0.33333334
+    ```
     """
 
     @typechecked

tensorflow_addons/metrics/multilabel_confusion_matrix.py

@@ -16,11 +16,10 @@
 
 import warnings
 
-import numpy as np
 import tensorflow as tf
-
 from tensorflow.keras import backend as K
 from tensorflow.keras.metrics import Metric
+import numpy as np
 
 from typeguard import typechecked
 from tensorflow_addons.utils.types import AcceptableDTypes, FloatTensorLike
@@ -47,36 +46,30 @@ class MultiLabelConfusionMatrix(Metric):
     - false negatives for class i in M(1,0)
     - true positives for class i in M(1,1)
 
-    Usage:
-
-    >>> y_true = tf.constant([[1, 0, 1], [0, 1, 0]], dtype=tf.int32)
-    >>> y_pred = tf.constant([[1, 0, 0],[0, 1, 1]], dtype=tf.int32)
-    >>> output1 = tfa.metrics.MultiLabelConfusionMatrix(num_classes=3)
-    >>> output1.update_state(y_true, y_pred)
-    >>> output1.result()
-    <tf.Tensor: shape=(3, 2, 2), dtype=float32, numpy=
-    array([[[1., 0.],
-            [0., 1.]],
-    <BLANKLINE>
-           [[1., 0.],
-            [0., 1.]],
-    <BLANKLINE>
-           [[0., 1.],
-            [1., 0.]]], dtype=float32)>
-    >>> y_true = tf.constant([[1, 0, 0], [0, 1, 0]], dtype=tf.int32)
-    >>> y_pred = tf.constant([[1, 0, 0],[0, 0, 1]], dtype=tf.int32)
-    >>> output2 = tfa.metrics.MultiLabelConfusionMatrix(num_classes=3)
-    >>> output2.update_state(y_true, y_pred)
-    >>> output2.result()
-    <tf.Tensor: shape=(3, 2, 2), dtype=float32, numpy=
-    array([[[1., 0.],
-            [0., 1.]],
-    <BLANKLINE>
-           [[1., 0.],
-            [1., 0.]],
-    <BLANKLINE>
-           [[1., 1.],
-            [0., 0.]]], dtype=float32)>
+    ```python
+    # multilabel confusion matrix
+    y_true = tf.constant([[1, 0, 1], [0, 1, 0]],
+             dtype=tf.int32)
+    y_pred = tf.constant([[1, 0, 0],[0, 1, 1]],
+             dtype=tf.int32)
+    output = MultiLabelConfusionMatrix(num_classes=3)
+    output.update_state(y_true, y_pred)
+    print('Confusion matrix:', output.result().numpy())
+
+    # Confusion matrix: [[[1 0] [0 1]] [[1 0] [0 1]]
+                      [[0 1] [1 0]]]
+
+    # if multiclass input is provided
+    y_true = tf.constant([[1, 0, 0], [0, 1, 0]],
+             dtype=tf.int32)
+    y_pred = tf.constant([[1, 0, 0],[0, 0, 1]],
+             dtype=tf.int32)
+    output = MultiLabelConfusionMatrix(num_classes=3)
+    output.update_state(y_true, y_pred)
+    print('Confusion matrix:', output.result().numpy())
+
+    # Confusion matrix: [[[1 0] [0 1]] [[1 0] [1 0]] [[1 1] [0 0]]]
+    ```
     """
 
     @typechecked

tensorflow_addons/metrics/r_square.py

@@ -16,7 +16,6 @@
 from typing import Tuple
 
 import tensorflow as tf
-
 from tensorflow.keras import backend as K
 from tensorflow.keras.metrics import Metric
 from tensorflow.python.ops import weights_broadcast_ops
@@ -62,13 +61,13 @@ class RSquare(Metric):
      of the same metric.
 
      Usage:
-
-    >>> actuals = tf.constant([1, 4, 3], dtype=tf.float32)
-    >>> preds = tf.constant([2, 4, 4], dtype=tf.float32)
-    >>> ans = tfa.metrics.RSquare()
-    >>> ans.update_state(actuals, preds)
-    >>> ans.result()
-    <tf.Tensor: shape=(), dtype=float32, numpy=0.57142854>
+     ```python
+     actuals = tf.constant([1, 4, 3], dtype=tf.float32)
+     preds = tf.constant([2, 4, 4], dtype=tf.float32)
+     result = tf.keras.metrics.RSquare()
+     result.update_state(actuals, preds)
+     print('R^2 score is: ', r1.result().numpy()) # 0.57142866
+    ```
     """
 
     @typechecked

tensorflow_addons/optimizers/lookahead.py

@@ -34,8 +34,10 @@ class Lookahead(tf.keras.optimizers.Optimizer):
 
     Example of usage:
 
-    >>> opt = tf.keras.optimizers.SGD(learning_rate=0.01)
-    >>> opt = tfa.optimizers.Lookahead(opt)
+    ```python
+    opt = tf.keras.optimizers.SGD(learning_rate)
+    opt = tfa.optimizers.Lookahead(opt)
+    ```
     """
 
     @typechecked