def getIterator(): dataDir = PathDefaults.getDataDir() + "cluster/" nbUser = 10000 # set to 'None' to have all users nbPurchasesPerIt = 500 # set to 'None' to take all the purchases per date startingIteration = 300 endingIteration = 600 # set to 'None' to have all iterations stepSize = 1 return itertools.islice(BemolData.getGraphIterator(dataDir, nbUser, nbPurchasesPerIt), startingIteration, endingIteration, stepSize)
def __init__(self): dataDir = PathDefaults.getDataDir() + "cluster/" nbUser = 2000 # set to 'None' to have all users nbPurchasesPerIt = 50 # set to 'None' to take all the purchases # per date startingIteration = 20 endingIteration = None # set to 'None' to have all iterations stepSize = 10 iterator = itertools.islice(BemolData.getGraphIterator(dataDir, nbUser, nbPurchasesPerIt), startingIteration, endingIteration, stepSize) self.iterator = iterator
def getIterator(): bemolIterator = BemolData.getGraphIterator(dataDir, dataArgs.nbUser, dataArgs.nbPurchasesPerIt) if dataArgs.maxComponents: bemolIterator = MaxComponentsIterator(bemolIterator, dataArgs.maxComponents) return itertools.islice(bemolIterator, dataArgs.startingIteration, dataArgs.endingIteration, dataArgs.stepSize)