From 74ba5b0bc1e73471c4ccc3fd0fda5d45ec2b3445 Mon Sep 17 00:00:00 2001 From: Av Shrikumar Date: Tue, 23 Oct 2018 11:42:38 -0700 Subject: [PATCH] added string decode for python 3 fix --- deeplift/conversion/kerasapi_conversion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deeplift/conversion/kerasapi_conversion.py b/deeplift/conversion/kerasapi_conversion.py index 2b4b9f2..48c2eda 100644 --- a/deeplift/conversion/kerasapi_conversion.py +++ b/deeplift/conversion/kerasapi_conversion.py @@ -404,7 +404,8 @@ def insert_weights_into_nested_model_config(nested_model_weights, layer_name = layer_config["config"]["name"] layer_weights = [np.array(nested_model_weights[x]) for x in nested_model_weights.keys() if - x.startswith(layer_name+"/")] + (x.decode("utf-8") if hasattr(x, 'decode') + else x).startswith(layer_name+"/")] if (len(layer_weights) > 0): layer_config["config"]["weights"] = layer_weights