add afm to ctr

apache · Jul 18, 2020 · a7da1c3 · a7da1c3
1 parent ab7b4b7
commit a7da1c3
Showing 1 changed file with 2 additions and 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
@@ -14,5 +14,6 @@
 # limitations under the License.
 
 from .deepfm import DeepFM
+from .afm import AttentionalFM  
 
-__all__ = ["DeepFM"]
+__all__ = ['DeepFM', 'AttentionalFM']