Пример #1
0
def testRecMethod(recType = 0):
  """Test single recommender.
     Input:
       recType - id of recommender.
     Output:
       None.
  """
  info = '############%s#############' % util.getMethodName(recType)
  start_time = time.time()
  songDict = persist.readSongFromFile()
  allPlaylist = persist.readPlaylistFromFile()
  recallTotal = 0.0
  precisionTotal = 0.0
  f1Total = 0.0
  maeTotal = 0.0
  rmseTotal = 0.0
  if recType == const.KNN:
    simFilename = '../txt/simMatrix.txt'
    pid2Index,index2Pid,countMatrix = util.getUserSongMatrix(allPlaylist,songDict)
    if os.path.exists(simFilename):
      print '......'
      simFile = open(simFilename,'r')
      line = simFile.readline().rstrip('\n')
      simMatrix = eval(line)
      print len(simMatrix)
      print len(simMatrix[1234])
      simFile.close()
    else:
      print '++++++'
      simMatrix = util.getUserSimMatrix(countMatrix,pid2Index)
  for scale in range(10):
    playlistDict = allPlaylist[scale]
    playlistDict = allPlaylist[scale]
    if recType == const.ARIMA:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.SIMILAR:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.AVG:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.ARIMA_SIMILAR:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.ARIMA_AVG:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.KNN:
      recDict = predict.getRecDictOfUserKNN(playlistDict,songDict,scale,pid2Index,countMatrix,simMatrix)
    elif recType == const.MARKOV:
      recDict = predict.getRecDictOfFirstMarkov(allPlaylist,songDict,scale)
    elif recType == const.MARKOV_3:
      recDict = predict.getRecDictOfThreeOrderMarkov(allPlaylist,songDict,scale)
    elif recType == const.PATTERN:
      recDict = predict.getRecDictOfMostPattern(allPlaylist,songDict,scale)
    recall,precision,f1 = util.getTopNIndex(recDict,playlistDict)
    mae,rmse = util.getMAEandRMSE(recDict,playlistDict,songDict)
    recallTotal += recall
    precisionTotal += precision
    f1Total += f1
    maeTotal += mae
    rmseTotal += rmse

  recall = recallTotal / 10
  precision = precisionTotal / 10
  f1 = f1Total / 10
  mae = maeTotal / 10
  rmse = rmseTotal / 10

  print info
  logging.info(info)
  print 'Recall = ',recall
  logging.info('Recall = %f' % recall)
  print 'Precision = ',precision
  logging.info('Precision = %f' % precision)
  print 'F1-Score = ',f1
  logging.info('F1-Score = %f' % f1)
  print 'MAE = ',mae
  logging.info('MAE = %f' % mae)
  print 'RMSE = ',rmse
  logging.info('RMSE = %f' % rmse)
  print 'Consumed: %ds' % (time.time()-start_time)
  logging.info('Consumed: %ds' % (time.time()-start_time))
Пример #2
0
def testRecMethod(recType=0):
    """Test single recommender.
     Input:
       recType - id of recommender.
     Output:
       None.
  """
    info = '############%s#############' % util.getMethodName(recType)
    start_time = time.time()
    songDict = persist.readSongFromFile()
    allPlaylist = persist.readPlaylistFromFile()
    recallTotal = 0.0
    precisionTotal = 0.0
    f1Total = 0.0
    maeTotal = 0.0
    rmseTotal = 0.0
    if recType == const.KNN:
        simFilename = '../txt/simMatrix.txt'
        pid2Index, index2Pid, countMatrix = util.getUserSongMatrix(
            allPlaylist, songDict)
        if os.path.exists(simFilename):
            print '......'
            simFile = open(simFilename, 'r')
            line = simFile.readline().rstrip('\n')
            simMatrix = eval(line)
            print len(simMatrix)
            print len(simMatrix[1234])
            simFile.close()
        else:
            print '++++++'
            simMatrix = util.getUserSimMatrix(countMatrix, pid2Index)
    for scale in range(10):
        playlistDict = allPlaylist[scale]
        playlistDict = allPlaylist[scale]
        if recType == const.ARIMA:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.SIMILAR:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.AVG:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.ARIMA_SIMILAR:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.ARIMA_AVG:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.KNN:
            recDict = predict.getRecDictOfUserKNN(playlistDict, songDict,
                                                  scale, pid2Index,
                                                  countMatrix, simMatrix)
        elif recType == const.MARKOV:
            recDict = predict.getRecDictOfFirstMarkov(allPlaylist, songDict,
                                                      scale)
        elif recType == const.MARKOV_3:
            recDict = predict.getRecDictOfThreeOrderMarkov(
                allPlaylist, songDict, scale)
        elif recType == const.PATTERN:
            recDict = predict.getRecDictOfMostPattern(allPlaylist, songDict,
                                                      scale)
        recall, precision, f1 = util.getTopNIndex(recDict, playlistDict)
        mae, rmse = util.getMAEandRMSE(recDict, playlistDict, songDict)
        recallTotal += recall
        precisionTotal += precision
        f1Total += f1
        maeTotal += mae
        rmseTotal += rmse

    recall = recallTotal / 10
    precision = precisionTotal / 10
    f1 = f1Total / 10
    mae = maeTotal / 10
    rmse = rmseTotal / 10

    print info
    logging.info(info)
    print 'Recall = ', recall
    logging.info('Recall = %f' % recall)
    print 'Precision = ', precision
    logging.info('Precision = %f' % precision)
    print 'F1-Score = ', f1
    logging.info('F1-Score = %f' % f1)
    print 'MAE = ', mae
    logging.info('MAE = %f' % mae)
    print 'RMSE = ', rmse
    logging.info('RMSE = %f' % rmse)
    print 'Consumed: %ds' % (time.time() - start_time)
    logging.info('Consumed: %ds' % (time.time() - start_time))
Пример #3
0
def getErrorOfRecMethod(recType = 0):
  """Get error of all recommender.
     Input:
       recType - id of recommender.
     Output:
       recalls,precisions,f1s,maes,rmses.
  """
  start_time = time.time()
  songDict = persist.readSongFromFile()
  allPlaylist = persist.readPlaylistFromFile()
  recalls = []
  precisions = []
  f1s = []
  maes = []
  rmses = []
  if recType == const.KNN:
    pid2Index,index2Pid,countMatrix = util.getUserSongMatrix(allPlaylist,songDict)
    simFilename = '../txt/simMatrix.txt'
    if os.path.exists(simFilename):
      print '......'
      simFile = open(simFilename,'r')
      line = simFile.readline().rstrip('\n')
      simMatrix = eval(line)
      print len(simMatrix)
      print len(simMatrix[1234])
      simFile.close()
    else:
      print '++++++'
      simMatrix = util.getUserSimMatrix(countMatrix,pid2Index)
  for scale in range(10):
    playlistDict = allPlaylist[scale]
    if recType == const.ARIMA:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.SIMILAR:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.AVG:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.ARIMA_SIMILAR:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.ARIMA_AVG:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.ALL_HYBRID:
      recDict = predict.getRecDict(playlistDict,songDict,recType,scale)
    elif recType == const.KNN:
      recDict = predict.getRecDictOfUserKNN(playlistDict,songDict,scale,pid2Index,countMatrix,simMatrix)
    elif recType == const.MARKOV:
      recDict = predict.getRecDictOfFirstMarkov(allPlaylist,songDict,scale)
    elif recType == const.MARKOV_3:
      recDict = predict.getRecDictOfThreeOrderMarkov(allPlaylist,songDict,scale)
    elif recType == const.PATTERN:
      recDict = predict.getRecDictOfMostPattern(allPlaylist,songDict,scale)
    index = 0
    for topN in range(1,const.TOP_N,1):
      recall,precision,f1 = util.getTopNIndex(recDict,playlistDict,topN)
      mae,rmse = util.getMAEandRMSE(recDict,playlistDict,songDict,topN)
      if scale == 0:
        recalls.append(recall)
        precisions.append(precision)
        f1s.append(f1)
        maes.append(mae)
        rmses.append(rmse)
      else:
        recalls[index] += recall
        precisions[index] += precision
        f1s[index] += f1
        maes[index] += mae
        rmses[index] += rmse
      index += 1

  #cal the avg value
  recalls = [recall / 10 for recall in recalls]
  precisions = [precision / 10 for precision in precisions]
  f1s = [f1 / 10 for f1 in f1s]
  maes = [mae / 10 for mae in maes]
  rmses = [rmse / 10 for rmse in rmses]

  #logging info to log
  index = 0
  for topN in range(1,const.TOP_N,1):
    print '%d:TopN = %d:%f %f %f %f %f' % (recType,topN,recalls[index],precisions[index],f1s[index],maes[index],rmses[index])
    logging.info('%d>%d:%f %f %f %f %f' % (recType,topN,recalls[index],precisions[index],f1s[index],maes[index],rmses[index]))
    index += 1
  end_time = time.time()
  print 'Consumed:%d' % (end_time-start_time)
  return recalls,precisions,f1s,maes,rmses  
Пример #4
0
def getErrorOfRecMethod(recType=0):
    """Get error of all recommender.
     Input:
       recType - id of recommender.
     Output:
       recalls,precisions,f1s,maes,rmses.
  """
    start_time = time.time()
    songDict = persist.readSongFromFile()
    allPlaylist = persist.readPlaylistFromFile()
    recalls = []
    precisions = []
    f1s = []
    maes = []
    rmses = []
    if recType == const.KNN:
        pid2Index, index2Pid, countMatrix = util.getUserSongMatrix(
            allPlaylist, songDict)
        simFilename = '../txt/simMatrix.txt'
        if os.path.exists(simFilename):
            print '......'
            simFile = open(simFilename, 'r')
            line = simFile.readline().rstrip('\n')
            simMatrix = eval(line)
            print len(simMatrix)
            print len(simMatrix[1234])
            simFile.close()
        else:
            print '++++++'
            simMatrix = util.getUserSimMatrix(countMatrix, pid2Index)
    for scale in range(10):
        playlistDict = allPlaylist[scale]
        if recType == const.ARIMA:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.SIMILAR:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.AVG:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.ARIMA_SIMILAR:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.ARIMA_AVG:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.ALL_HYBRID:
            recDict = predict.getRecDict(playlistDict, songDict, recType,
                                         scale)
        elif recType == const.KNN:
            recDict = predict.getRecDictOfUserKNN(playlistDict, songDict,
                                                  scale, pid2Index,
                                                  countMatrix, simMatrix)
        elif recType == const.MARKOV:
            recDict = predict.getRecDictOfFirstMarkov(allPlaylist, songDict,
                                                      scale)
        elif recType == const.MARKOV_3:
            recDict = predict.getRecDictOfThreeOrderMarkov(
                allPlaylist, songDict, scale)
        elif recType == const.PATTERN:
            recDict = predict.getRecDictOfMostPattern(allPlaylist, songDict,
                                                      scale)
        index = 0
        for topN in range(1, const.TOP_N, 1):
            recall, precision, f1 = util.getTopNIndex(recDict, playlistDict,
                                                      topN)
            mae, rmse = util.getMAEandRMSE(recDict, playlistDict, songDict,
                                           topN)
            if scale == 0:
                recalls.append(recall)
                precisions.append(precision)
                f1s.append(f1)
                maes.append(mae)
                rmses.append(rmse)
            else:
                recalls[index] += recall
                precisions[index] += precision
                f1s[index] += f1
                maes[index] += mae
                rmses[index] += rmse
            index += 1

    #cal the avg value
    recalls = [recall / 10 for recall in recalls]
    precisions = [precision / 10 for precision in precisions]
    f1s = [f1 / 10 for f1 in f1s]
    maes = [mae / 10 for mae in maes]
    rmses = [rmse / 10 for rmse in rmses]

    #logging info to log
    index = 0
    for topN in range(1, const.TOP_N, 1):
        print '%d:TopN = %d:%f %f %f %f %f' % (recType, topN, recalls[index],
                                               precisions[index], f1s[index],
                                               maes[index], rmses[index])
        logging.info('%d>%d:%f %f %f %f %f' %
                     (recType, topN, recalls[index], precisions[index],
                      f1s[index], maes[index], rmses[index]))
        index += 1
    end_time = time.time()
    print 'Consumed:%d' % (end_time - start_time)
    return recalls, precisions, f1s, maes, rmses