Skip to content

Commit 1404f0d

Browse files
author
浅梦
authored
Update interaction.py
1 parent 37eae73 commit 1404f0d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

deepctr/layers/interaction.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1039,6 +1039,6 @@ def compute_output_shape(self, input_shape):
10391039
return (None, 1, filed_size*(filed_size-1)//2 * embedding_size)
10401040

10411041
def get_config(self, ):
1042-
config = {'type': self.bilinear_type, 'seed': self.seed}
1042+
config = {'bilinear_type': self.bilinear_type, 'seed': self.seed}
10431043
base_config = super(BilinearInteraction, self).get_config()
1044-
return dict(list(base_config.items()) + list(config.items()))
1044+
return dict(list(base_config.items()) + list(config.items()))

0 commit comments

Comments
 (0)