xb = x_train[start_i:end_i]
yb = y_train[start_i:end_i]