Пример #1
0
def test_get_from_module_uses_custom_object():
    get_custom_objects().clear()
    assert (get_from_module("CustomObject", globals(), "test_generic_utils") == CustomObject)
    with pytest.raises(ValueError):
        get_from_module("TestObject", globals(), "test_generic_utils")
    with custom_object_scope({"TestObject": CustomObject}):
        assert (get_from_module("TestObject", globals(), "test_generic_utils") == CustomObject)
Пример #2
0
def test_get_from_module_uses_custom_object():
    get_custom_objects().clear()
    assert (get_from_module("CustomObject", globals(),
                            "test_generic_utils") == CustomObject)
    with pytest.raises(ValueError):
        get_from_module("TestObject", globals(), "test_generic_utils")
    with custom_object_scope({"TestObject": CustomObject}):
        assert (get_from_module("TestObject", globals(),
                                "test_generic_utils") == CustomObject)
Пример #3
0
def get(identifier, kwargs=None):
    if K.backend() == 'tensorflow':
        # Wrap TF optimizer instances
        if isinstance(identifier, tf.train.Optimizer):
            return TFOptimizer(identifier)
    # Instantiate a Keras optimizer
    return get_from_module(identifier, globals(), 'optimizer',
                           instantiate=True, kwargs=kwargs)
Пример #4
0
def get(identifier, kwargs=None):
    return get_from_module(identifier, globals(), 'similarity',
                           instantiate=False, kwargs=kwargs)
def get(identifier, kwargs=None):
    return get_from_module(identifier,
                           globals(),
                           'regularizer',
                           instantiate=True,
                           kwargs=kwargs)
Пример #6
0
def get(identifier, **kwargs):
    return get_from_module(identifier, globals(),
                           'initialization', kwargs=kwargs)
Пример #7
0
def get(identifier, kwargs=None):
    return get_from_module(identifier, globals(), 'optimizer',
                           instantiate=True, kwargs=kwargs)
Пример #8
0
def get(identifier):
    return get_from_module(identifier, globals(), 'fusions')
Пример #9
0
def get(identifier, **kwargs):
    return get_from_module(identifier,
                           globals(),
                           'initialization',
                           kwargs=kwargs)
Пример #10
0
def get(identifier):
    return get_from_module(identifier, globals(), 'objective')
Пример #11
0
def get(identifier):
    return get_from_module(identifier, globals(), 'metric')
Пример #12
0
def get(identifier):
    return get_from_module(identifier, globals(), 'policy', instantiate=True)
Пример #13
0
def get(identifier):
    return get_from_module(identifier, globals(), 'policy', instantiate=True)