def testFindAllSimilarityFromNodeOnPathSimExampleThree(self):
"""
Tests similarity for all other nodes given a signle node, using example 3 from PathSim paper
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
strategy = PageRankStrategy(graph)
mike = authorMap['Mike']
mostSimilarNodes = strategy.findMostSimilarNodes(mike, 1)
self.assertEquals([authorMap['Ann']], mostSimilarNodes)
def testFindAllSimilarityFromNodeOnPathSimExampleThree(self):
"""
Tests similarity for all other nodes given a signle node, using example 3 from PathSim paper
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree(
)
strategy = PageRankStrategy(graph)
mike = authorMap['Mike']
mostSimilarNodes = strategy.findMostSimilarNodes(mike, 1)
self.assertEquals([authorMap['Ann']], mostSimilarNodes)
def testFindAllSimilarityFromNodeOnPathSimExampleThree(self):
"""
Tests similarity for all other nodes given a single node, using example 3 from PathSim paper
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
metaPath = [Author, Paper, Conference, Paper, Author]
strategy = PathSimStrategy(graph, metaPath)
mike = authorMap['Mike']
mostSimilarNodes = strategy.findMostSimilarNodes(mike, 5)
self.assertEquals([authorMap['Bob'], authorMap['Mary'], authorMap['Jim']], mostSimilarNodes)
def testFindAllSimilarityFromNodeOnPathSimExampleThree(self):
"""
Tests similarity for all other nodes given a single node, using example 3 from PathSim paper
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree(
)
metaPath = [Author, Paper, Conference, Paper, Author]
strategy = PathSimStrategy(graph, metaPath)
mike = authorMap['Mike']
mostSimilarNodes = strategy.findMostSimilarNodes(mike, 5)
self.assertEquals(
[authorMap['Bob'], authorMap['Mary'], authorMap['Jim']],
mostSimilarNodes)
def testFindSingleSimilarityPathSimExampleThree(self):
"""
Tests pairwise similarity for nodes, using example 3 from PathSim paper (compute similarity scores from Mike)
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
metaPath = [Author, Paper, Conference, Paper, Author]
strategy = PathSimStrategy(graph, metaPath)
mike = authorMap['Mike']
jimScore, maryScore, bobScore, annScore = strategy.findSimilarityScores(
mike, [authorMap['Jim'], authorMap['Mary'], authorMap['Bob'], authorMap['Ann']]
)
self.assertEquals(bobScore, max([jimScore, maryScore, bobScore, annScore]))
self.assertEquals(annScore, 0)
def testFindSingleSimilarityPathSimExampleThree(self):
"""
Tests pairwise similarity for nodes, using example 3 from PathSim paper (compute similarity scores from Mike)
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
strategy = PageRankStrategy(graph)
mike = authorMap['Mike']
jimScore = strategy.findSimilarityScore(mike, authorMap['Jim'])
maryScore = strategy.findSimilarityScore(mike, authorMap['Mary'])
bobScore = strategy.findSimilarityScore(mike, authorMap['Bob'])
annScore = strategy.findSimilarityScore(mike, authorMap['Ann'])
self.assertTrue(annScore >= maryScore)
self.assertTrue(annScore >= jimScore)
self.assertTrue(annScore >= bobScore)
def testFindSingleSimilarityPathSimExampleThree(self):
"""
Tests pairwise similarity for nodes, using example 3 from PathSim paper (compute similarity scores from Mike)
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree(
)
strategy = PageRankStrategy(graph)
mike = authorMap['Mike']
jimScore = strategy.findSimilarityScore(mike, authorMap['Jim'])
maryScore = strategy.findSimilarityScore(mike, authorMap['Mary'])
bobScore = strategy.findSimilarityScore(mike, authorMap['Bob'])
annScore = strategy.findSimilarityScore(mike, authorMap['Ann'])
self.assertTrue(annScore >= maryScore)
self.assertTrue(annScore >= jimScore)
self.assertTrue(annScore >= bobScore)
def testFindSingleSimilarityPathSimExampleThree(self):
"""
Tests pairwise similarity for nodes, using example 3 from PathSim paper (compute similarity scores from Mike)
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree(
)
metaPath = [Author, Paper, Conference, Paper, Author]
strategy = PathSimStrategy(graph, metaPath)
mike = authorMap['Mike']
jimScore, maryScore, bobScore, annScore = strategy.findSimilarityScores(
mike, [
authorMap['Jim'], authorMap['Mary'], authorMap['Bob'],
authorMap['Ann']
])
self.assertEquals(bobScore,
max([jimScore, maryScore, bobScore, annScore]))
self.assertEquals(annScore, 0)
def testConstructPathSimExampleThree(self):
"""
Tests the construction of "Example 3" from PathSim paper. Specifically, checks adjacency matrix shown in
this example for Author-Paper-Conference meta paths.
"""
graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
metaPath = [Author, Paper, Conference]
metaPathUtility = EdgeBasedMetaPathUtility()
# Mike's adjacency to conferences
self.assertEquals(2, len(metaPathUtility.findMetaPaths(graph, authorMap['Mike'], conferenceMap['SIGMOD'], metaPath)))
self.assertEquals(1, len(metaPathUtility.findMetaPaths(graph, authorMap['Mike'], conferenceMap['VLDB'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Mike'], conferenceMap['ICDE'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Mike'], conferenceMap['KDD'], metaPath)))
# Jim's adjacency to conferences
self.assertEquals(50, len(metaPathUtility.findMetaPaths(graph, authorMap['Jim'], conferenceMap['SIGMOD'], metaPath)))
self.assertEquals(20, len(metaPathUtility.findMetaPaths(graph, authorMap['Jim'], conferenceMap['VLDB'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Jim'], conferenceMap['ICDE'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Jim'], conferenceMap['KDD'], metaPath)))
# Mary's adjacency to conferences
self.assertEquals(2, len(metaPathUtility.findMetaPaths(graph, authorMap['Mary'], conferenceMap['SIGMOD'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Mary'], conferenceMap['VLDB'], metaPath)))
self.assertEquals(1, len(metaPathUtility.findMetaPaths(graph, authorMap['Mary'], conferenceMap['ICDE'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Mary'], conferenceMap['KDD'], metaPath)))
# Bob's adjacency to conferences
self.assertEquals(2, len(metaPathUtility.findMetaPaths(graph, authorMap['Bob'], conferenceMap['SIGMOD'], metaPath)))
self.assertEquals(1, len(metaPathUtility.findMetaPaths(graph, authorMap['Bob'], conferenceMap['VLDB'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Bob'], conferenceMap['ICDE'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Bob'], conferenceMap['KDD'], metaPath)))
# Ann's adjacency to conferences
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Ann'], conferenceMap['SIGMOD'], metaPath)))
self.assertEquals(0, len(metaPathUtility.findMetaPaths(graph, authorMap['Ann'], conferenceMap['VLDB'], metaPath)))
self.assertEquals(1, len(metaPathUtility.findMetaPaths(graph, authorMap['Ann'], conferenceMap['ICDE'], metaPath)))
self.assertEquals(1, len(metaPathUtility.findMetaPaths(graph, authorMap['Ann'], conferenceMap['KDD'], metaPath)))
def run(self):
citationMap = {
'Mike': {
'Mike': 0,
'Jim': 0,
'Mary': 0,
'Bob': 0,
'Ann': 0,
'Joe': 0,
'Nancy': 0
},
'Jim': {
'Mike': 20,
'Jim': 0,
'Mary': 20,
'Bob': 20,
'Ann': 0,
'Joe': 20,
'Nancy': 0
},
'Mary': {
'Mike': 1,
'Jim': 10,
'Mary': 0,
'Bob': 1,
'Ann': 0,
'Joe': 1,
'Nancy': 0
},
'Bob': {
'Mike': 1,
'Jim': 10,
'Mary': 1,
'Bob': 0,
'Ann': 0,
'Joe': 1,
'Nancy': 0
},
'Ann': {
'Mike': 0,
'Jim': 0,
'Mary': 0,
'Bob': 0,
'Ann': 0,
'Joe': 0,
'Nancy': 0
},
'Joe': {
'Mike': 0,
'Jim': 0,
'Mary': 0,
'Bob': 0,
'Ann': 0,
'Joe': 0,
'Nancy': 0
},
'Nancy': {
'Mike': 1,
'Jim': 10,
'Mary': 1,
'Bob': 1,
'Ann': 0,
'Joe': 1,
'Nancy': 0
}
}
self.graph, authorMap, conferenceMap =\
SampleGraphUtility.constructPathSimExampleThree(extraAuthorsAndCitations=True, citationMap = citationMap)
# Get the nodes we care about
conferences = [
conferenceMap['SIGMOD'], conferenceMap['VLDB'],
conferenceMap['ICDE'], conferenceMap['KDD']
]
authors = [
authorMap['Mike'],
authorMap['Jim'],
authorMap['Mary'],
authorMap['Bob'],
authorMap['Ann'],
authorMap['Joe'],
authorMap['Nancy'],
]
metaPathUtility = EdgeBasedMetaPathUtility()
# Project a 2-typed heterogeneous graph over adapted PathSim example
publicationProjectedGraph = metaPathUtility.createHeterogeneousProjection(
self.graph, [Author, Paper, Conference], symmetric=True)
self.output('\nAdjacency Matrix (Projected):')
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [conference.name for conference in conferences]]
rows += [[author.name] + [
publicationProjectedGraph.getNumberOfEdges(author, conference)
for conference in conferences
] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Project a homogeneous citation graph over adapted PathSim example
citationProjectedGraph = metaPathUtility.createHomogeneousProjection(
self.graph, [Author, Paper, Paper, Author])
self.output('\nCitation Matrix:')
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [author.name for author in authors]]
rows += [[author.name] + [
citationProjectedGraph.getNumberOfEdges(author, otherAuthor)
for otherAuthor in authors
] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Output total out/in citations
self.output('\nCitations Total:')
totalCitationsTable = texttable.Texttable()
rows = [['Author', 'In', 'Out']]
for author in authors:
inCount = sum(
citationProjectedGraph.getNumberOfEdges(otherAuthor, author)
for otherAuthor in authors)
outCount = sum(
citationProjectedGraph.getNumberOfEdges(author, otherAuthor)
for otherAuthor in authors)
rows += [[author.name, inCount, outCount]]
totalCitationsTable.add_rows(rows)
self.output(totalCitationsTable.draw())
# Get PathSim similarity scores
pathSimStrategy = PathSimStrategy(
self.graph, [Author, Paper, Conference, Paper, Author], True)
self.outputSimilarityScores(authorMap, authors, pathSimStrategy,
'APCPA PathSim')
# Output SimRank-related scores
strategy = SimRankStrategy(self.graph, [Author, Paper, Paper, Author],
symmetric=True)
self.outputSimilarityScores(authorMap, authors, strategy, "SimRank")
# Output the projected PageRank/HITS similarity scores
for name, algorithm in zip(
['PageRank', 'HITS'],
[PageRankDistanceStrategy, HITSDistanceStrategy]):
strategy = algorithm(self.graph, [Author, Paper, Paper, Author],
symmetric=True)
self.outputSimilarityScores(authorMap, authors, strategy,
"%s-Distance" % name)
# Get NeighborSim similarity scores
inNeighborSimStrategy = NeighborSimStrategy(
self.graph, [Author, Paper, Paper, Author])
self.outputSimilarityScores(authorMap, authors, inNeighborSimStrategy,
'APPA NeighborSim-In')
outNeighborSimStrategy = NeighborSimStrategy(
self.graph, [Author, Paper, Paper, Author],
reversed=True,
smoothed=True)
self.outputSimilarityScores(authorMap, authors, outNeighborSimStrategy,
'APPA NeighborSim-Out')
# Combined best PR-distance algorithm
simRankStrategy = SimRankStrategy(self.graph,
[Author, Paper, Paper, Author],
symmetric=True)
simRank = AggregateSimilarityStrategy(
self.graph, [pathSimStrategy, simRankStrategy], [0.5, 0.5])
self.outputSimilarityScores(authorMap, authors, simRank,
'APCPA Pathsim, APPA SimRank')
# Combined best neighborsim score
combinedNeighborSim = AggregateSimilarityStrategy(
self.graph,
[pathSimStrategy, inNeighborSimStrategy, outNeighborSimStrategy],
[0.6, 0.2, 0.2])
self.outputSimilarityScores(
authorMap, authors, combinedNeighborSim,
'APCPA Pathsim, APPA NeighborSim-Combined')
def run(self):
citationMap = {
'Mike': {'Mike': 0, 'Jim': 0, 'Mary': 0, 'Bob': 0, 'Ann': 0, 'Joe': 0, 'Nancy': 0},
'Jim': {'Mike': 20, 'Jim': 0, 'Mary': 20, 'Bob': 20, 'Ann': 0, 'Joe': 20, 'Nancy': 0},
'Mary': {'Mike': 1, 'Jim': 10, 'Mary': 0, 'Bob': 1, 'Ann': 0, 'Joe': 1, 'Nancy': 0},
'Bob': {'Mike': 1, 'Jim': 10, 'Mary': 1, 'Bob': 0, 'Ann': 0, 'Joe': 1, 'Nancy': 0},
'Ann': {'Mike': 0, 'Jim': 0, 'Mary': 0, 'Bob': 0, 'Ann': 0, 'Joe': 0, 'Nancy': 0},
'Joe': {'Mike': 0, 'Jim': 0, 'Mary': 0, 'Bob': 0, 'Ann': 0, 'Joe': 0, 'Nancy': 0},
'Nancy': {'Mike': 1, 'Jim': 10, 'Mary': 1, 'Bob': 1, 'Ann': 0, 'Joe': 1, 'Nancy': 0}
}
self.graph, authorMap, conferenceMap =\
SampleGraphUtility.constructPathSimExampleThree(extraAuthorsAndCitations=True, citationMap = citationMap)
# Get the nodes we care about
conferences = [
conferenceMap['SIGMOD'],
conferenceMap['VLDB'],
conferenceMap['ICDE'],
conferenceMap['KDD']
]
authors = [
authorMap['Mike'],
authorMap['Jim'],
authorMap['Mary'],
authorMap['Bob'],
authorMap['Ann'],
authorMap['Joe'],
authorMap['Nancy'],
]
metaPathUtility = EdgeBasedMetaPathUtility()
# Project a 2-typed heterogeneous graph over adapted PathSim example
publicationProjectedGraph = metaPathUtility.createHeterogeneousProjection(self.graph, [Author, Paper, Conference], symmetric = True)
self.output('\nAdjacency Matrix (Projected):')
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [conference.name for conference in conferences]]
rows += [[author.name] + [publicationProjectedGraph.getNumberOfEdges(author, conference) for conference in conferences] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Project a homogeneous citation graph over adapted PathSim example
citationProjectedGraph = metaPathUtility.createHomogeneousProjection(self.graph, [Author, Paper, Paper, Author])
self.output('\nCitation Matrix:')
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [author.name for author in authors]]
rows += [[author.name] + [citationProjectedGraph.getNumberOfEdges(author, otherAuthor) for otherAuthor in authors] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Output total out/in citations
self.output('\nCitations Total:')
totalCitationsTable = texttable.Texttable()
rows = [['Author', 'In', 'Out']]
for author in authors:
inCount = sum(citationProjectedGraph.getNumberOfEdges(otherAuthor, author) for otherAuthor in authors)
outCount = sum(citationProjectedGraph.getNumberOfEdges(author, otherAuthor) for otherAuthor in authors)
rows += [[author.name, inCount, outCount]]
totalCitationsTable.add_rows(rows)
self.output(totalCitationsTable.draw())
# Get PathSim similarity scores
pathSimStrategy = PathSimStrategy(self.graph, [Author, Paper, Conference, Paper, Author], True)
self.outputSimilarityScores(authorMap, authors, pathSimStrategy, 'APCPA PathSim')
# Output SimRank-related scores
strategy = SimRankStrategy(self.graph, [Author, Paper, Paper, Author], symmetric=True)
self.outputSimilarityScores(authorMap, authors, strategy, "SimRank")
# Output the projected PageRank/HITS similarity scores
for name, algorithm in zip(['PageRank', 'HITS'], [PageRankDistanceStrategy, HITSDistanceStrategy]):
strategy = algorithm(self.graph, [Author, Paper, Paper, Author], symmetric=True)
self.outputSimilarityScores(authorMap, authors, strategy, "%s-Distance" % name)
# Get NeighborSim similarity scores
inNeighborSimStrategy = NeighborSimStrategy(self.graph, [Author, Paper, Paper, Author])
self.outputSimilarityScores(authorMap, authors, inNeighborSimStrategy, 'APPA NeighborSim-In')
outNeighborSimStrategy = NeighborSimStrategy(self.graph, [Author, Paper, Paper, Author], reversed=True, smoothed=True)
self.outputSimilarityScores(authorMap, authors, outNeighborSimStrategy, 'APPA NeighborSim-Out')
# Combined best PR-distance algorithm
simRankStrategy = SimRankStrategy(self.graph, [Author, Paper, Paper, Author], symmetric=True)
simRank = AggregateSimilarityStrategy(self.graph, [pathSimStrategy, simRankStrategy], [0.5, 0.5])
self.outputSimilarityScores(authorMap, authors, simRank, 'APCPA Pathsim, APPA SimRank')
# Combined best neighborsim score
combinedNeighborSim = AggregateSimilarityStrategy(self.graph, [pathSimStrategy, inNeighborSimStrategy, outNeighborSimStrategy], [0.6, 0.2, 0.2])
self.outputSimilarityScores(authorMap, authors, combinedNeighborSim, 'APCPA Pathsim, APPA NeighborSim-Combined')
def run(self):
self.graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree()
# Get the nodes we care about
conferences = [
conferenceMap['SIGMOD'],
conferenceMap['VLDB'],
conferenceMap['ICDE'],
conferenceMap['KDD']
]
authors = [
authorMap['Mike'],
authorMap['Jim'],
authorMap['Mary'],
authorMap['Bob'],
authorMap['Ann'],
]
metaPathUtility = EdgeBasedMetaPathUtility()
self.output('\nAPC Adjacency Matrix:')
apcadjMatrix, nodesIndex = metaPathUtility.getAdjacencyMatrixFromGraph(self.graph, [Author, Paper, Conference], project=True)
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [conference.name for conference in conferences]]
rows += [[author.name] + [apcadjMatrix[nodesIndex[author]][nodesIndex[conference]] for conference in conferences] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
self.output('\nCPA Adjacency Matrix:')
cpaadjMatrix, dsad = metaPathUtility.getAdjacencyMatrixFromGraph(self.graph, [Conference, Paper, Author], project=True)
adjMatrixTable = texttable.Texttable()
rows = [['Conference'] + [author.name for author in authors]]
rows += [[conference.name] + [cpaadjMatrix[nodesIndex[conference]][nodesIndex[author]] for author in authors] for conference in conferences]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
self.output('\nAPCPA Adjacency Matrix (Computed):')
adjMatrix = numpy.dot(apcadjMatrix, cpaadjMatrix)
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [author.name for author in authors]]
rows += [[author.name] + [adjMatrix[nodesIndex[author]][nodesIndex[otherAuthor]] for otherAuthor in authors] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Output homogeneous simrank comparison
homogeneousSimRankStrategy = SimRankStrategy(self.graph)
self.outputSimilarityScores(authorMap, authors, homogeneousSimRankStrategy, 'Homogeneous SimRank')
projectedGraph = metaPathUtility.createHeterogeneousProjection(self.graph, [Author, Paper, Conference], symmetric = True)
# Output heterogeneous simrank comparison
heterogeneousSimRankStrategy = SimRankStrategy(projectedGraph)
self.outputSimilarityScores(authorMap, authors, heterogeneousSimRankStrategy, 'APC Heterogeneous SimRank')
# Output heterogeneous simrank w/ squared neighbors comparison
def sqNeighborsNorm(graph, a, b, sim):
aNeighbors, bNeighbors = graph.getPredecessors(a), graph.getPredecessors(b)
return float(len(aNeighbors)**2 * len(bNeighbors)**2)
heterogeneousSquaredSimRankStrategy = SimRankStrategy(projectedGraph, normalization=sqNeighborsNorm)
self.outputSimilarityScores(authorMap, authors, heterogeneousSquaredSimRankStrategy, 'Squared Heterogeneous SimRank')
# Output NeighborSim similarity scores
neighborSimStrategy = NeighborSimStrategy(self.graph, [Author, Paper, Conference], symmetric=True)
self.outputSimilarityScores(authorMap, authors, neighborSimStrategy, 'APC NeighborSim')
# Output the PathSim similarity scores
pathsimStrategy = PathSimStrategy(self.graph, [Author, Paper, Conference, Paper, Author], symmetric=True)
self.outputSimilarityScores(authorMap, authors, pathsimStrategy, 'APCPA PathSim')
def run(self):
self.graph, authorMap, conferenceMap = SampleGraphUtility.constructPathSimExampleThree(
)
# Get the nodes we care about
conferences = [
conferenceMap['SIGMOD'], conferenceMap['VLDB'],
conferenceMap['ICDE'], conferenceMap['KDD']
]
authors = [
authorMap['Mike'],
authorMap['Jim'],
authorMap['Mary'],
authorMap['Bob'],
authorMap['Ann'],
]
metaPathUtility = EdgeBasedMetaPathUtility()
self.output('\nAPC Adjacency Matrix:')
apcadjMatrix, nodesIndex = metaPathUtility.getAdjacencyMatrixFromGraph(
self.graph, [Author, Paper, Conference], project=True)
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [conference.name for conference in conferences]]
rows += [[author.name] + [
apcadjMatrix[nodesIndex[author]][nodesIndex[conference]]
for conference in conferences
] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
self.output('\nCPA Adjacency Matrix:')
cpaadjMatrix, dsad = metaPathUtility.getAdjacencyMatrixFromGraph(
self.graph, [Conference, Paper, Author], project=True)
adjMatrixTable = texttable.Texttable()
rows = [['Conference'] + [author.name for author in authors]]
rows += [[conference.name] + [
cpaadjMatrix[nodesIndex[conference]][nodesIndex[author]]
for author in authors
] for conference in conferences]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
self.output('\nAPCPA Adjacency Matrix (Computed):')
adjMatrix = numpy.dot(apcadjMatrix, cpaadjMatrix)
adjMatrixTable = texttable.Texttable()
rows = [['Author'] + [author.name for author in authors]]
rows += [[author.name] + [
adjMatrix[nodesIndex[author]][nodesIndex[otherAuthor]]
for otherAuthor in authors
] for author in authors]
adjMatrixTable.add_rows(rows)
self.output(adjMatrixTable.draw())
# Output homogeneous simrank comparison
homogeneousSimRankStrategy = SimRankStrategy(self.graph)
self.outputSimilarityScores(authorMap, authors,
homogeneousSimRankStrategy,
'Homogeneous SimRank')
projectedGraph = metaPathUtility.createHeterogeneousProjection(
self.graph, [Author, Paper, Conference], symmetric=True)
# Output heterogeneous simrank comparison
heterogeneousSimRankStrategy = SimRankStrategy(projectedGraph)
self.outputSimilarityScores(authorMap, authors,
heterogeneousSimRankStrategy,
'APC Heterogeneous SimRank')
# Output heterogeneous simrank w/ squared neighbors comparison
def sqNeighborsNorm(graph, a, b, sim):
aNeighbors, bNeighbors = graph.getPredecessors(
a), graph.getPredecessors(b)
return float(len(aNeighbors)**2 * len(bNeighbors)**2)
heterogeneousSquaredSimRankStrategy = SimRankStrategy(
projectedGraph, normalization=sqNeighborsNorm)
self.outputSimilarityScores(authorMap, authors,
heterogeneousSquaredSimRankStrategy,
'Squared Heterogeneous SimRank')
# Output NeighborSim similarity scores
neighborSimStrategy = NeighborSimStrategy(self.graph,
[Author, Paper, Conference],
symmetric=True)
self.outputSimilarityScores(authorMap, authors, neighborSimStrategy,
'APC NeighborSim')
# Output the PathSim similarity scores
pathsimStrategy = PathSimStrategy(
self.graph, [Author, Paper, Conference, Paper, Author],
symmetric=True)
self.outputSimilarityScores(authorMap, authors, pathsimStrategy,
'APCPA PathSim')
