Ejemplos de assert_tensors en Python

Ejemplo n.º 1

    def assert_equal_metrics(m1, m2):
        assert_tensors(m1.pop('input_layer', None),
                       m2.pop('input_layer', None))
        assert_tensors(m1.pop('output_layer', None),
                       m2.pop('output_layer', None))

        assert_equal_dict(m1, m2)

Ejemplo n.º 2

    def assert_equal_losses(l1, l2):
        assert_tensors(l1.pop('input_layer', None),
                       l2.pop('input_layer', None))
        assert_tensors(l1.pop('output_layer', None),
                       l2.pop('output_layer', None))

        assert_equal_dict(l1, l2)