The TopographicSimilarity class wasn't using the functions sender_input_distance_fn and message_distance_fn given in __init__. In addition, it couldn't work for sender_inputs of more than 1D (e.g., images), because the Spearman correlation function expects 2D input. Fixed both in the print_message method.
Description
Added self.sender_input_distance_fn and self.message_distance_fn to the self.compute_topsim call inside the print_message method.
Flattened sender_input to be 2D inside the print_message method.
The TopographicSimilarity class wasn't using the functions
sender_input_distance_fn
andmessage_distance_fn
given in__init__
. In addition, it couldn't work forsender_input
s of more than 1D (e.g., images), because the Spearman correlation function expects 2D input. Fixed both in theprint_message
method.Description
self.sender_input_distance_fn
andself.message_distance_fn
to theself.compute_topsim
call inside theprint_message
method.sender_input
to be 2D inside theprint_message
method.Related Issue
https://github.com/facebookresearch/EGG/issues/250