Skip to content

Instantly share code, notes, and snippets.

@timehaven
Created July 19, 2017 15:47
Show Gist options
  • Select an option

  • Save timehaven/ef82ef489987e66debead3e3c9013c95 to your computer and use it in GitHub Desktop.

Select an option

Save timehaven/ef82ef489987e66debead3e3c9013c95 to your computer and use it in GitHub Desktop.

Revisions

  1. timehaven created this gist Jul 19, 2017.
    12 changes: 12 additions & 0 deletions mm1.py
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,12 @@
    ...
    i, j = 0, batch_size
    for _ in range(nbatches):
    sub = df.iloc[i:j]
    X2 = bc[i:j]
    ...
    # Calculate X and Y appropriately
    ...
    yield [X, X2], Y
    i = j
    j += batch_size
    ...