From a7da1c3487783cb7a9933192bc862d78015b718d Mon Sep 17 00:00:00 2001 From: Andrew Hsieh Date: Sun, 12 Jul 2020 20:22:13 +0800 Subject: [PATCH] add afm to ctr --- .../pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py index 3fb493537e..ac6fd3f574 100644 --- a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py +++ b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py @@ -14,5 +14,6 @@ # limitations under the License. from .deepfm import DeepFM +from .afm import AttentionalFM -__all__ = ["DeepFM"] +__all__ = ['DeepFM', 'AttentionalFM']