jfae

from random import random

def crpgen(N = None, alpha = 1.0):
  """
  A generator that implements the Chinese Restaurant Process
  """
  counts = []
  n = 0
  while N == None or n < N:
    # Compute the (unnormalized) probabilities of assigning the new object