Update interaction.py

shenweichen · Jul 10, 2019 · 1404f0d · 1404f0d
1 parent 37eae73
commit 1404f0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepctr/layers/interaction.py b/deepctr/layers/interaction.py
@@ -1039,6 +1039,6 @@ def compute_output_shape(self, input_shape):
         return (None, 1, filed_size*(filed_size-1)//2 * embedding_size)
 
     def get_config(self, ):
-        config = {'type': self.bilinear_type, 'seed': self.seed}
+        config = {'bilinear_type': self.bilinear_type, 'seed': self.seed}
         base_config = super(BilinearInteraction, self).get_config()
-        return dict(list(base_config.items()) + list(config.items()))
+        return dict(list(base_config.items()) + list(config.items()))