bug_fix

keras-team · Sep 17, 2019 · 0173bbe · 0173bbe
1 parent 17221fa
commit 0173bbe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/autokeras/hypermodel/graph.py b/autokeras/hypermodel/graph.py
@@ -219,12 +219,11 @@ def _add_node(self, input_node):
             self._node_to_id[input_node] = len(self._node_to_id)
 
     def _get_metrics(self):
-        metrics = []
-        for metrics_list in [output_node.in_blocks[0].metrics for
-                             output_node in self.outputs
-                             if isinstance(output_node.in_blocks[0],
-                                           head.Head)]:
-            metrics += metrics_list
+        metrics = {}
+        for output_node in self.outputs:
+            block = output_node.in_blocks[0]
+            if isinstance(block, head.Head):
+                metrics[block.output_name] = block.metrics
         return metrics
 
     def _get_loss(self):

diff --git a/autokeras/hypermodel/head.py b/autokeras/hypermodel/head.py
@@ -148,6 +148,7 @@ def __init__(self,
         if not self.metrics:
             self.metrics = ['mean_squared_error']
         self.dropout_rate = dropout_rate
+        self.output_name = self.name + '_output'
 
     @property
     def loss(self):
@@ -173,5 +174,5 @@ def build(self, hp, inputs=None):
         if dropout_rate > 0:
             output_node = tf.keras.layers.Dropout(dropout_rate)(output_node)
         output_node = block.Flatten().build(hp, output_node)
-        output_node = tf.keras.layers.Dense(self.output_shape[-1])(output_node)
+        output_node = tf.keras.layers.Dense(self.output_shape[-1], name=self.output_name)(output_node)
         return output_node