We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 37eae73 commit 1404f0dCopy full SHA for 1404f0d
deepctr/layers/interaction.py
@@ -1039,6 +1039,6 @@ def compute_output_shape(self, input_shape):
1039
return (None, 1, filed_size*(filed_size-1)//2 * embedding_size)
1040
1041
def get_config(self, ):
1042
- config = {'type': self.bilinear_type, 'seed': self.seed}
+ config = {'bilinear_type': self.bilinear_type, 'seed': self.seed}
1043
base_config = super(BilinearInteraction, self).get_config()
1044
- return dict(list(base_config.items()) + list(config.items()))
+ return dict(list(base_config.items()) + list(config.items()))
0 commit comments