generate_batch

def generate_batches(X, y, batch_size):
    """
    param X: np.array[n_objects, n_features] --- матрица объекты-признаки
    param y: np.array[n_objects] --- вектор целевых переменных
    """
    assert len(X) == len(y)
    np.random.seed(42)
    X = np.array(X)
    y = np.array(y)
    X_p = []
    y_p = []
    perm = np.random.permutation(len(X))
    for i in range(len(X)):
      X_p.append(X[perm[i]])
      y_p.append(y[perm[i]])
    X_b = X_p[:batch_size]
    y_b = y_p[:batch_size]
    for i in range(len(X) // batch_size):
      X_b = X_p[i * batch_size: i * batch_size + batch_size]
      y_b = y_p[i * batch_size: i * batch_size + batch_size]
      yield X_b, y_b