updated experiment.py in sync with mulitlabel case