#!/usr/bin/python import math movieFileName = 'movieCorrProbeResults.txt' userFileName = 'userCorrProbeResults2.txt' newPredictionFileName = '/home/gregstoll/stuff/netflixprize/download/dummyoutput.txt' def average(mFN, uFN, oFN): mF = open(mFN, 'r') uF = open(uFN, 'r') oF = open(oFN, 'w') mStr = mF.readline() uStr = uF.readline() while (uStr != ''): mStr = mStr.strip() uStr = uStr.strip() if (mStr[len(mStr)-1:] == ':'): if (uStr[len(uStr)-1:] != ':'): print "Line mismatch! mStr=%s, uStr=%s" % (mStr, uStr) oF.write(mStr + "\n") else: mNum = float(mStr) uNum = float(uStr) oNum = .75 * mNum + .25 * uNum oF.write(str(oNum) + "\n") mStr = mF.readline() uStr = uF.readline() if (mStr != ''): print "Warning: fewer lines in user file than movie file!!" #print "found %d ratings (should be 1408395)" % count mF.close() uF.close() oF.close() if (__name__ == '__main__'): average(movieFileName, userFileName, newPredictionFileName)