Skip to content

Commit df06f25

Browse files
author
shenweichen
committed
no message
1 parent 2d72040 commit df06f25

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

deepctr/inputs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
from collections import OrderedDict, namedtuple, defaultdict
1010
from itertools import chain
11+
from copy import copy
1112

1213
from tensorflow.python.keras.initializers import RandomNormal
1314
from tensorflow.python.keras.layers import Embedding, Input, Flatten
@@ -177,7 +178,7 @@ def create_embedding_matrix(feature_columns, l2_reg, init_std, seed, prefix="",
177178

178179
def get_linear_logit(features, feature_columns, units=1, use_bias=False, init_std=0.0001, seed=1024, prefix='linear',
179180
l2_reg=0):
180-
linear_feature_columns = feature_columns.copy()
181+
linear_feature_columns = copy(feature_columns)
181182
for i in range(len(linear_feature_columns)):
182183
if isinstance(linear_feature_columns[i], SparseFeat):
183184
linear_feature_columns[i] = linear_feature_columns[i]._replace(embedding_dim=1)

deepctr/layers/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def compute_mask(self, inputs, mask):
122122
return None
123123

124124
def get_config(self, ):
125-
config = {'mode': self.mode, 'l2_reg': self.l2_reg}
125+
config = {'mode': self.mode, 'l2_reg': self.l2_reg,'use_bias':self.use_bias}
126126
base_config = super(Linear, self).get_config()
127127
return dict(list(base_config.items()) + list(config.items()))
128128

0 commit comments

Comments
 (0)