def test_weights_are_correct_sizes(self):
# pylint: disable=protected-access
similarity = MultiHeadedSimilarity(num_heads=3, tensor_1_dim=9, tensor_1_projected_dim=6,
tensor_2_dim=6, tensor_2_projected_dim=12)
assert list(similarity._tensor_1_projection.size()) == [9, 6]
assert list(similarity._tensor_2_projection.size()) == [6, 12]
with pytest.raises(ConfigurationError):
similarity = MultiHeadedSimilarity(num_heads=3, tensor_1_dim=10)
with pytest.raises(ConfigurationError):
params = Params({'num_heads': 3, 'tensor_1_dim': 9, 'tensor_2_dim': 10})
MultiHeadedSimilarity.from_params(params)
def test_weights_are_correct_sizes(self):
# pylint: disable=protected-access
similarity = MultiHeadedSimilarity(num_heads=3,
tensor_1_dim=9,
tensor_1_projected_dim=6,
tensor_2_dim=6,
tensor_2_projected_dim=12)
assert list(similarity._tensor_1_projection.size()) == [9, 6]
assert list(similarity._tensor_2_projection.size()) == [6, 12]
with pytest.raises(ConfigurationError):
similarity = MultiHeadedSimilarity(num_heads=3, tensor_1_dim=10)
with pytest.raises(ConfigurationError):
params = Params({
'num_heads': 3,
'tensor_1_dim': 9,
'tensor_2_dim': 10
})
MultiHeadedSimilarity.from_params(params)
def test_weights_are_correct_sizes(self):
similarity = MultiHeadedSimilarity(
num_heads=3,
tensor_1_dim=9,
tensor_1_projected_dim=6,
tensor_2_dim=6,
tensor_2_projected_dim=12,
)
assert list(similarity._tensor_1_projection.size()) == [9, 6]
assert list(similarity._tensor_2_projection.size()) == [6, 12]
with pytest.raises(ConfigurationError):
similarity = MultiHeadedSimilarity(num_heads=3, tensor_1_dim=10)
with pytest.raises(ConfigurationError):
params = Params({
"num_heads": 3,
"tensor_1_dim": 9,
"tensor_2_dim": 10
})
MultiHeadedSimilarity.from_params(params)
