fixed some bugs with attribute training return data