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__)