Skip to content

Commit 16092f2

Browse files
committed
Compatible with TF2.3 #25
1 parent 7ab834e commit 16092f2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

deeptables/models/deepmodel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,8 @@ def __init__(self,
3939
self.model_file = model_file
4040
self.model = None
4141
if model_file is not None:
42-
self.model = load_model(model_file, dt_custom_objects) # fixme `load_model` executed multiple times in a process, resulting in a metric name rename to like auc_1, auc_2
42+
self.model = load_model(model_file,
43+
dt_custom_objects) # fixme `load_model` executed multiple times in a process, resulting in a metric name rename to like auc_1, auc_2
4344

4445
def fit(self, X=None, y=None, batch_size=128, epochs=1, verbose=1, callbacks=None,
4546
validation_split=0.2, validation_data=None, shuffle=True,
@@ -293,7 +294,7 @@ def __build_denses(self, continuous_columns, continuous_inputs, dense_dropout, u
293294
dense_layer = None
294295
if continuous_inputs:
295296
if len(continuous_inputs) > 1:
296-
dense_layer = Concatenate(name=consts.LAYER_NAME_CONCAT_CONT_INPUTS)(continuous_inputs.values())
297+
dense_layer = Concatenate(name=consts.LAYER_NAME_CONCAT_CONT_INPUTS)(list(continuous_inputs.values()))
297298
else:
298299
dense_layer = list(continuous_inputs.values())[0]
299300
if dense_dropout > 0:
@@ -425,4 +426,3 @@ def __getitem__(self, item):
425426
if not isinstance(item, str):
426427
raise KeyError(f"Key should be str but is {item}")
427428
return self.data[item.lower()]
428-

0 commit comments

Comments
 (0)