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]
