def loadVSMatrices(prefix = 'cmj', loadTFIDF = True, loadLSI = True):
    """load and return vector space matrices of the documents"""
    global vecTFIDF, vecLSI
    vecTFIDF, vecLSI = None, None
    if loadTFIDF:
        vecTFIDF = matutils.loadMatrix(common.matrixFile(prefix + 'TFIDF.mm'))
    if loadLSI:
        vecLSI = matutils.loadMatrix(common.matrixFile(prefix + 'LSI_VT.mm'))
    return vecTFIDF, vecLSI
def loadVSMatrices(prefix='cmj', loadTFIDF=True, loadLSI=True):
    """load and return vector space matrices of the documents"""
    global vecTFIDF, vecLSI
    vecTFIDF, vecLSI = None, None
    if loadTFIDF:
        vecTFIDF = matutils.loadMatrix(common.matrixFile(prefix + 'TFIDF.mm'))
    if loadLSI:
        vecLSI = matutils.loadMatrix(common.matrixFile(prefix + 'LSI_VT.mm'))
    return vecTFIDF, vecLSI
def buildDocDoc(arts, type, language):
    ipyutils.loadDicts(prefix="gensim_" + language)
    arts = [art for art in arts if art.id_int in ipyutils.rdocids]
    assert len(arts) == len(ipyutils.rdocids)

    logging.info("loading msc<->id mapping")
    cats, rcats = loadMsc2Id(language)

    mscsFile = common.matrixFile("mscs_%s.mm" % type)
    matMsc = matutils.loadMatrix(mscsFile)
    mscDict = {}
    for art in arts:
        artId = ipyutils.rdocids[art.id_int]
        mscIds = [rcats[mscs.niceMSC(msc)[0]] for msc in art.msc]
        mscDict[artId] = mscIds

    logging.info("computing doc*doc similarity matrix based on %s" % mscsFile)
    docdoc = numpy.zeros((len(arts), len(arts)), numpy.float32)

    for i in xrange(len(arts)):
        if i % 100 == 0:
            logging.info("PROGRESS: %i/%i" % (i, len(arts)))
        art1Id = ipyutils.rdocids[arts[i].id_int]
        for j in xrange(i, len(arts)):
            art2Id = ipyutils.rdocids[arts[j].id_int]
            bestScore = 0.0
            for msc1Id in mscDict[art1Id]:
                for msc2Id in mscDict[art2Id]:
                    bestScore = max(bestScore, matMsc[msc1Id, msc2Id])
            docdoc[art1Id, art2Id] = docdoc[art2Id, art1Id] = bestScore

    matutils.saveMatrix(docdoc, common.matrixFile("docdoc_" + language + "_%s.mm" % type), sparse=False)
    return docdoc
Example #4
0
def buildDocDoc(arts, type, language):
    ipyutils.loadDicts(prefix='gensim_' + language)
    arts = [art for art in arts if art.id_int in ipyutils.rdocids]
    assert (len(arts) == len(ipyutils.rdocids))

    logging.info("loading msc<->id mapping")
    cats, rcats = loadMsc2Id(language)

    mscsFile = common.matrixFile("mscs_%s.mm" % type)
    matMsc = matutils.loadMatrix(mscsFile)
    mscDict = {}
    for art in arts:
        artId = ipyutils.rdocids[art.id_int]
        mscIds = [rcats[mscs.niceMSC(msc)[0]] for msc in art.msc]
        mscDict[artId] = mscIds

    logging.info("computing doc*doc similarity matrix based on %s" % mscsFile)
    docdoc = numpy.zeros((len(arts), len(arts)), numpy.float32)

    for i in xrange(len(arts)):
        if i % 100 == 0:
            logging.info("PROGRESS: %i/%i" % (i, len(arts)))
        art1Id = ipyutils.rdocids[arts[i].id_int]
        for j in xrange(i, len(arts)):
            art2Id = ipyutils.rdocids[arts[j].id_int]
            bestScore = 0.0
            for msc1Id in mscDict[art1Id]:
                for msc2Id in mscDict[art2Id]:
                    bestScore = max(bestScore, matMsc[msc1Id, msc2Id])
            docdoc[art1Id, art2Id] = docdoc[art2Id, art1Id] = bestScore

    matutils.saveMatrix(docdoc,
                        common.matrixFile("docdoc_" + language +
                                          "_%s.mm" % type),
                        sparse=False)
    return docdoc
        breaks.append((i, msc))
        oldMsc = msc
logging.info("%i breaks" % len(breaks))
print "breaks:", breaks

# print individual category matrices
breaks.append((i, msc))  # append the last category
print "breaks with end:", breaks

new2old = {}
for i, (msc, id) in enumerate(art2msc):
    new2old[i] = ipyutils.rdocids[id]
print '==first 100 new2old:', new2old.items()[:100]

logging.info("loading cossim matrix")
mat = matutils.loadMatrix(common.matrixFile("gensim_eng%ssim.mm" % (SIM_TYPE)))
for i in xrange(len(breaks) - 1):
    numArts = breaks[i + 1][0] - breaks[i][0]
    logging.debug("%i articles in category %s" % (numArts, breaks[i][1]))
    matId = numpy.zeros((numArts, numArts), numpy.float)
    for i1 in xrange(numArts):
        print breaks[i][0] + i1, i1, art2mscOld[breaks[i][0] + i1]
        for i2 in xrange(numArts):
            pos1, pos2 = breaks[i][0] + i1, breaks[i][0] + i2
            matId[i1, i2] = mat[new2old[pos1], new2old[pos2]]
    processMsc(matId, breaks[i][1][0])

sys.exit(0)

new2old = {}
for i, (msc, id) in enumerate(art2msc):
def loadConcepts(prefix='cmj'):
    """load and return the LSI_U matrix of concepts"""
    global concepts
    concepts = matutils.loadMatrix(common.matrixFile(prefix + 'LSI_U.mm'))
    return concepts
def loadSimMatrices(prefix='cmj'):
    global m, l
    m = matutils.loadMatrix(common.matrixFile(prefix + 'TFIDFsim.mm'))
    l = matutils.loadMatrix(common.matrixFile(prefix + 'LSIsim.mm'))
    return m, l
Example #8
0
def rmseFile(matfile1, matfile2):
    data1 = matutils.loadMatrix(common.matrixFile(matfile1))
    data2 = matutils.loadMatrix(common.matrixFile(matfile2))
    return matutils.rmse(data1, data2)
Example #9
0
def buildMscCentroidMatrix(language):
    logging.info("building MSC centroid matrix from %s" % ARTS_FILE)
    arts = [
        art for art in docsim.getArts(
            ARTS_FILE, acceptNoBody=False, acceptNoMsc=False)
        if art.language == language or language == 'any'
    ]
    prefix = 'mscs_serial_%s_' % language
    matFile = common.matrixFile(prefix + 'TFIDF_T.mm')
    if os.path.exists(matFile):
        logging.warning(
            'SKIPPING creating TFIDF matrix for %s (file %s present). Is this what you wanted?'
            % (language, matFile))
        tfidf = matutils.loadMatrix(matFile).tocsr()
    else:
        logging.info('creating TFIDF matrix for %s to %s' %
                     (language, matFile))
        tfidf = docsim.buildTFIDFMatrices(arts,
                                          prefix=prefix,
                                          saveMatrices=False).tocsr()

    ipyutils.loadDicts(prefix=prefix)
    arts = [
        art for art in arts if art.id_int in ipyutils.rdocids
    ]  # remove articles that had empty body (according to their tfidf vector)
    if len(ipyutils.rdocids) != len(arts):
        logging.error(
            "no. of TFIDF document = %i, but there are %i documents in the database (mismatch)"
            % (len(ipyutils.rdocids), len(arts)))
        raise Exception(
            "different size of database/dictionary; version mismatch?")

    cats, rcats = loadMsc2Id(language)  # from buildPure
    #    print "mscs:", cats

    logging.info("loading tfidf collection matrix (for centroids)")
    tfidf = matutils.loadMatrix(
        common.matrixFile('gensim_' + language + 'TFIDF_T.mm')).tocsr()
    logging.debug("loaded %ix%i matrix" % tfidf.shape)

    logging.info("computing centroids")
    centroids = numpy.zeros((len(cats), tfidf.shape[1]), numpy.float)
    #    print "centroids.shape =", centroids.shape
    num = numpy.zeros((len(cats), ), numpy.int)
    artCnt = 0
    for art in arts:
        if not art.id_int in ipyutils.rdocids:
            logging.warning("article not found among docids: %s" % art)
            continue
        artCnt += 1
        artId = ipyutils.rdocids[art.id_int]
        tops = [mscs.niceMSC(msc)[0] for msc in art.msc]
        tops = set(
            tops
        )  # only count each top-level once (comment out this line to count e.g. 30H55 and 30.13 twice for this article, as cat. 30)
        for top in tops:
            mscId = rcats[top]
            vec = tfidf[artId].toarray()
            vec.shape = (vec.size, )
            #            print "vec.shape = ", vec.shape
            centroids[mscId] += vec
            num[mscId] += 1
        if artCnt < 10 or artCnt % 1000 == 0:
            logging.debug(
                "sanity check - article %s has id %i and has mscs=%s, mscsIds=%s"
                % (art.id_int, artId, art.msc,
                   [rcats[mscs.niceMSC(msc)[0]] for msc in art.msc]))
    if not artCnt == tfidf.shape[0]:
        raise Exception("not all articles used; database/matrix mismatch?")
    for i, vec in enumerate(centroids):
        logging.info(
            "centroid for msc %s (id %i) is an average of %i vectors" %
            (cats[i], i, num[i]))
        if numpy.sum(numpy.abs(vec)) == 0:
            logging.warning("empty centroid for msc %s (msc int id %i)" %
                            (cats[i], i))
    for mscId in cats.iterkeys():
        centroids[mscId] /= num[mscId]
    logging.info(
        "used %i articles for %i vectors (articles may have more than one msc and so can be counted more than once)"
        % (artCnt, sum(num)))

    logging.info("computing MSC centroid matrix")
    resultCentroid = numpy.zeros((len(cats), len(cats)), dtype=numpy.float32)
    for idi, cati in cats.iteritems():
        for idj, catj in cats.iteritems():
            #            print idi, cati, idj, catj
            sim = matutils.cossim(centroids[idi], centroids[idj])
            if numpy.isfinite(sim):
                resultCentroid[idi, idj] = sim
            else:
                resultCentroid[idi, idj] = 0.0

    matutils.saveMatrix(resultCentroid,
                        common.matrixFile("mscs_centroid_%s.mm" % language),
                        sparse=False)
		oldMsc = msc
logging.info("%i breaks" % len(breaks))
print "breaks:", breaks

# print individual category matrices
breaks.append((i, msc)) # append the last category
print "breaks with end:", breaks

new2old = {}
for i, (msc, id) in enumerate(art2msc):
    new2old[i] = ipyutils.rdocids[id]
print '==first 100 new2old:', new2old.items()[:100]
	

logging.info("loading cossim matrix")
mat = matutils.loadMatrix(common.matrixFile("gensim_eng%ssim.mm" % (SIM_TYPE)))
for i in xrange(len(breaks) - 1):
	numArts = breaks[i + 1][0] - breaks[i][0]
	logging.debug("%i articles in category %s" % (numArts, breaks[i][1]))
	matId = numpy.zeros((numArts, numArts), numpy.float)
	for i1 in xrange(numArts):
		print breaks[i][0] + i1, i1, art2mscOld[breaks[i][0] + i1]
		for i2 in xrange(numArts):
			pos1, pos2 = breaks[i][0] + i1, breaks[i][0] + i2
			matId[i1, i2] = mat[new2old[pos1], new2old[pos2]]
	processMsc(matId, breaks[i][1][0])

sys.exit(0)

new2old = {}
for i, (msc, id) in enumerate(art2msc):
Example #11
0
#!/usr/bin/env python2.5

import logging

logging.root.level = 10

import common
import pylab
import numpy
import matutils

mat = matutils.loadMatrix(common.matrixFile("gensim_engTFIDFsim.mm"))
#mat = numpy.zeros((10, 20), float)
#mat[2] = 0.2
#mat[8] = 0.9
logging.info("%ix%i matrix loaded" % mat.shape)
pylab.figure()
pylab.imshow(mat, cmap = pylab.cm.gray, interpolation = "nearest")
pylab.savefig("tfidf_sim")
def rmseFile(matfile1, matfile2):
    data1 = matutils.loadMatrix(common.matrixFile(matfile1))
    data2 = matutils.loadMatrix(common.matrixFile(matfile2))
    return matutils.rmse(data1, data2)
def buildMscCentroidMatrix(language):
    logging.info("building MSC centroid matrix from %s" % ARTS_FILE)
    arts = [
        art
        for art in docsim.getArts(ARTS_FILE, acceptNoBody=False, acceptNoMsc=False)
        if art.language == language or language == "any"
    ]
    prefix = "mscs_serial_%s_" % language
    matFile = common.matrixFile(prefix + "TFIDF_T.mm")
    if os.path.exists(matFile):
        logging.warning(
            "SKIPPING creating TFIDF matrix for %s (file %s present). Is this what you wanted?" % (language, matFile)
        )
        tfidf = matutils.loadMatrix(matFile).tocsr()
    else:
        logging.info("creating TFIDF matrix for %s to %s" % (language, matFile))
        tfidf = docsim.buildTFIDFMatrices(arts, prefix=prefix, saveMatrices=False).tocsr()

    ipyutils.loadDicts(prefix=prefix)
    arts = [
        art for art in arts if art.id_int in ipyutils.rdocids
    ]  # remove articles that had empty body (according to their tfidf vector)
    if len(ipyutils.rdocids) != len(arts):
        logging.error(
            "no. of TFIDF document = %i, but there are %i documents in the database (mismatch)"
            % (len(ipyutils.rdocids), len(arts))
        )
        raise Exception("different size of database/dictionary; version mismatch?")

    cats, rcats = loadMsc2Id(language)  # from buildPure
    #    print "mscs:", cats

    logging.info("loading tfidf collection matrix (for centroids)")
    tfidf = matutils.loadMatrix(common.matrixFile("gensim_" + language + "TFIDF_T.mm")).tocsr()
    logging.debug("loaded %ix%i matrix" % tfidf.shape)

    logging.info("computing centroids")
    centroids = numpy.zeros((len(cats), tfidf.shape[1]), numpy.float)
    #    print "centroids.shape =", centroids.shape
    num = numpy.zeros((len(cats),), numpy.int)
    artCnt = 0
    for art in arts:
        if not art.id_int in ipyutils.rdocids:
            logging.warning("article not found among docids: %s" % art)
            continue
        artCnt += 1
        artId = ipyutils.rdocids[art.id_int]
        tops = [mscs.niceMSC(msc)[0] for msc in art.msc]
        tops = set(
            tops
        )  # only count each top-level once (comment out this line to count e.g. 30H55 and 30.13 twice for this article, as cat. 30)
        for top in tops:
            mscId = rcats[top]
            vec = tfidf[artId].toarray()
            vec.shape = (vec.size,)
            #            print "vec.shape = ", vec.shape
            centroids[mscId] += vec
            num[mscId] += 1
        if artCnt < 10 or artCnt % 1000 == 0:
            logging.debug(
                "sanity check - article %s has id %i and has mscs=%s, mscsIds=%s"
                % (art.id_int, artId, art.msc, [rcats[mscs.niceMSC(msc)[0]] for msc in art.msc])
            )
    if not artCnt == tfidf.shape[0]:
        raise Exception("not all articles used; database/matrix mismatch?")
    for i, vec in enumerate(centroids):
        logging.info("centroid for msc %s (id %i) is an average of %i vectors" % (cats[i], i, num[i]))
        if numpy.sum(numpy.abs(vec)) == 0:
            logging.warning("empty centroid for msc %s (msc int id %i)" % (cats[i], i))
    for mscId in cats.iterkeys():
        centroids[mscId] /= num[mscId]
    logging.info(
        "used %i articles for %i vectors (articles may have more than one msc and so can be counted more than once)"
        % (artCnt, sum(num))
    )

    logging.info("computing MSC centroid matrix")
    resultCentroid = numpy.zeros((len(cats), len(cats)), dtype=numpy.float32)
    for idi, cati in cats.iteritems():
        for idj, catj in cats.iteritems():
            #            print idi, cati, idj, catj
            sim = matutils.cossim(centroids[idi], centroids[idj])
            if numpy.isfinite(sim):
                resultCentroid[idi, idj] = sim
            else:
                resultCentroid[idi, idj] = 0.0

    matutils.saveMatrix(resultCentroid, common.matrixFile("mscs_centroid_%s.mm" % language), sparse=False)
def loadConcepts(prefix = 'cmj'):
    """load and return the LSI_U matrix of concepts"""
    global concepts
    concepts = matutils.loadMatrix(common.matrixFile(prefix + 'LSI_U.mm'))
    return concepts
def loadSimMatrices(prefix = 'cmj'):
    global m, l
    m = matutils.loadMatrix(common.matrixFile(prefix + 'TFIDFsim.mm'))
    l = matutils.loadMatrix(common.matrixFile(prefix + 'LSIsim.mm'))
    return m, l