Skip to content

Commit

Permalink
Implement base compute metrics
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanradev93 committed Jun 12, 2024
1 parent 8f195c2 commit 84ed906
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion bayesflow/experimental/networks/summary_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,4 +6,10 @@

class SummaryNetwork(keras.Layer):
def compute_metrics(self, data: dict[str, Tensor], stage: str = "training") -> dict[str, Tensor]:
raise NotImplementedError

# TODO - find a good way to deal with summary conditions
data["summary_outputs"] = self(
data["summary_variables"], training=stage == "training"
)
# TODO metrics
return {}

0 comments on commit 84ed906

Please sign in to comment.