No module named metric_learn

[ghost]

thanks for your work, when I use softmax example, I find this error"Traceback (most recent call last):
File "softmax_loss.py", line 12, in
from reid import datasets
File "/home/yuepan/code/pytorch_open_reid/reid/init.py", line 7, in
from . import metric_learning
File "/home/yuepan/code/pytorch_open_reid/reid/metric_learning/init.py", line 3, in
from metric_learn import (ITML_Supervised, LMNN, LSML_Supervised,
ImportError: No module named metric_learn"
and I find no metric_learn.py in folder

[zydou]

You should install metric-learn first.

pip install metric-learn

[ghost]

thx

[Cysu]

@zydou Thank you very much!