pytorch-metric-learning icon indicating copy to clipboard operation
pytorch-metric-learning copied to clipboard

Allow dim reducer to be passed in to BaseTester

Open KevinMusgrave opened this issue 4 years ago • 0 comments

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.

KevinMusgrave avatar May 29 '20 11:05 KevinMusgrave