def test__initialize_params_(self):
preheated_embeddings = np.array([
np.array([-1.0, -1.0, -6.0]),
np.array([-1.0, -1.0, 5.0]),
np.array([-1.0, -1.0, 4.0]),
np.array([-1.0, 3.0, -1.0]),
np.array([1.0, 1.0, -1.0]),
np.array([2.0, 1.0, -1.0]),
np.array([1.0, -1.0, 1.0]),
np.array([1.0, -1.0, -1.0])
])
interaction_sparse_tensor = interaction_map.idxs_to_tf([2])
test_netowrk = Network(cf, interaction_map, preheated_embeddings)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
result = sess.run(
test_netowrk.embedd_interaction_sparse_tensor(
interaction_sparse_tensor))
np.testing.assert_array_equal(result, [[-1.0, -1.0, 4.0]])
