Skip to content

Commit

Permalink
Compatible with TF2.3 #25
Browse files Browse the repository at this point in the history
  • Loading branch information
jackguagua committed Aug 12, 2020
1 parent 7ab834e commit 16092f2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions deeptables/models/deepmodel.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,8 @@ def __init__(self,
self.model_file = model_file
self.model = None
if model_file is not None:
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
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

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

0 comments on commit 16092f2

Please sign in to comment.