google-research / disentanglement_lib

disentanglement_lib is an open-source library for research on learning disentangled representations.
Apache License 2.0
1.38k stars 205 forks source link

Disentanglement metrics don't seem to work with continuous factors #25

Open asross opened 4 years ago

asross commented 4 years ago

I might be misunderstanding (e.g. perhaps all the ground-truth factors are meant to be discretized), but it seems like all of the metrics require latent factors to be discrete, e.g. using a classifier rather than a regressor for DCI. This is a bit inconvenient for users who would like to use your library to evaluate disentanglement solely given matrices of continuous latents and ground truth factors (without using the representation_function API). I know it's a design choice rather than a bug, but I'd still like to make the suggestion that it would be helpful for future users if you supported this use case!