def makeQuizArrays(qualfile): """ Given the text quiz file, make rating arrays """ from flixdata import readProbeFile qualdata = readProbeFile(qualfile) qualdata[:, 0] -= 1 # get the original user IDs origUserIDs = N.fromfile(origuserIDfile, dtype='int32') convert = N.r_[0:N.max(origUserIDs) + 1] convert[origUserIds] = N.r_[0:480189] qualdata[:, 1] = convert[qualdata[:, 1]]
def makeQuizArrays(qualfile): """ Given the text quiz file, make rating arrays """ from flixdata import readProbeFile qualdata = readProbeFile(qualfile) qualdata[:,0] -= 1 # get the original user IDs origUserIDs = N.fromfile(origuserIDfile, dtype='int32') convert = N.r_[0:N.max(origUserIDs)+1] convert[origUserIds] = N.r_[0:480189] qualdata[:,1] = convert[qualdata[:,1]] # etc etc
def readtestfile(probefile): """ mids, uids = readtestfile('probe.txt', fd) Read a txt probe file and convert the movie and user IDs to use a zero-index. We require a FlixData class instance (fd) because it's the easiest way (in my current setup) to zero out user IDs. """ from flixdata import FlixData, readProbeFile fd = FlixData(loaddata=False) data = readProbeFile(probefile) print('Re-indexing the data file...') data[:,0] -= 1 # zero-index the movies data[:,1] = fd.zeroIndexUserIDs(data[:,1]) return data[:,0], data[:,1]