train diversity