6 3 2 5000 1 2.0 (array([0, 1, 2], dtype=int8),) (array([1, 1, 1, 2, 2, 2], dtype=int8), array([0, 1, 2, 0, 1, 2], dtype=int8)) (array([1], dtype=int8),) (array([3], dtype=int8),)