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)