Imported from https://github.com/all-umass/metric_learn