Add option in LMNN to learn a diagonal transformation.