Added "one_hot" option for labels