pytorch-metric-learning
pytorch-metric-learning copied to clipboard
Allow dim reducer to be passed in to BaseTester
Currently there is the option to reduce the dimensionality of the embeddings using PCA. To allow more flexibility, this should be changed to a "dim reducer" input, which is expected to have a fit_transform function that can be applied to the embeddings before accuracy is computed.