[ML] Train - fix for virtual root