upgrade zero inflated lognormal loss, support export structure path, upgrade pre-commit hooks

Author: yangxudong

.github/workflows/code_style.yml

@@ -16,12 +16,6 @@ jobs:
           ref: ${{ github.event.pull_request.head.sha }}
           submodules: recursive
 
-      - name: Clean pre-commit
-        run: |
-          pre-commit --version
-          pre-commit clean || true
-          pre-commit gc || true
-
       - name: RunCiTest
         id: run_ci_test
         env:

.pre-commit-config.yaml

@@ -6,11 +6,7 @@ repos:
         additional_dependencies: [
           'flake8-docstrings==1.5.0'
         ]
-  - repo: https://github.com/asottile/seed-isort-config
-    rev: v2.2.0
-    hooks:
-      - id: seed-isort-config
-  - repo: https://github.com/timothycrosley/isort
+  - repo: https://github.com/pycqa/isort
     rev: 4.3.21
     hooks:
       - id: isort

easy_rec/python/model/deepfm.py

@@ -49,17 +49,25 @@ def build_input_layer(self, model_config, feature_configs):
     if not has_final:
       assert model_config.deepfm.wide_output_dim == model_config.num_class
     self._wide_output_dim = model_config.deepfm.wide_output_dim
-    if self._wide_output_dim != 1:
+    if self._wide_output_dim != self._num_class:
       logging.warning(
         'wide_output_dim not equal to 1, it is not a standard model'
       )
     super(DeepFM, self).build_input_layer(model_config, feature_configs)
 
   def build_predict_graph(self):
     # Wide
-    wide_fea = tf.reduce_sum(
-      self._wide_features, axis=1, keepdims=True, name='wide_feature'
-    )
+    if self._num_class > 1 and self._wide_output_dim == self._num_class:
+      wide_shape = tf.shape(self._wide_features)
+      new_shape = tf.stack(
+        [-1, wide_shape[1] // self._num_class, self._num_class]
+      )
+      wide_fea = tf.reshape(self._wide_features, new_shape)
+      wide_fea = tf.reduce_sum(wide_fea, axis=1, name='wide_feature')
+    else:
+      wide_fea = tf.reduce_sum(
+        self._wide_features, axis=1, keepdims=True, name='wide_feature'
+      )
 
     # FM
     fm_fea = fm.FM(name='fm_feature')(self._fm_features)