def __init__(self, device, encoder_type, decoder_type, input_size, hidden_size, **kwargs):
super().__init__()
self.encoder = EncoderLinearQuery(device=device, input_size=input_size, hidden_size=hidden_size,
**params_for('encoder', kwargs))
self.decoder = DecoderKeyValueGCNMultiEmbedding(device=device, hidden_size=hidden_size,
**params_for('decoder', kwargs))
self.device = device
def __init__(self, device, input_size, hidden_size, **kwargs):
super().__init__()
self.encoder = EncoderRNNQuery(device=device,
input_size=input_size,
hidden_size=hidden_size,
**params_for('encoder', kwargs))
self.decoder = DecoderMultiHopKeyValueSeparate(device=device,
hidden_size=hidden_size,
**params_for(
'decoder', kwargs))
self.device = device
def test_params_for(self, params_for, prefix, kwargs, expected):
assert params_for(prefix, kwargs) == expected
def _get_params_for(self, prefix):
return params_for(prefix, self.__dict__)
def test_params_for(self, params_for, prefix, kwargs, expected):
assert params_for(prefix, kwargs) == expected
