擅长:python、mysql、java
<p>在中添加以下函数AttentionWithContext.py文件:</p>
<pre><code>def create_custom_objects():
instance_holder = {"instance": None}
class ClassWrapper(AttentionWithContext):
def __init__(self, *args, **kwargs):
instance_holder["instance"] = self
super(ClassWrapper, self).__init__(*args, **kwargs)
def loss(*args):
method = getattr(instance_holder["instance"], "loss_function")
return method(*args)
def accuracy(*args):
method = getattr(instance_holder["instance"], "accuracy")
return method(*args)
return {"ClassWrapper": ClassWrapper ,"AttentionWithContext": ClassWrapper, "loss": loss,
"accuracy":accuracy}
</code></pre>
<p>加载模型时:</p>
^{pr2}$