class CommunityBuilderTest(unittest.TestCase):
def setUp(self):
self.cb = CommunityBuilder('../data/cn-topic.db')
def test_load_title_wordnet(self):
g = self.cb.load_title_wordnet()
print 'count of vertex and eges: %d, %d' %(g.vcount(),g.ecount())
def test_title_cluster(dbfile = '../data/sample_test.db'):
from CommunityBuilder import CommunityBuilder
import comdect
cb = CommunityBuilder(dbfile)
g = cb.load_title_wordnet(2)
detect = comdect.WalkCommunityDetection(min_nodes=15)
coms = detect.detect(g)
print len(coms)
coms = comuid2name(g,coms)
rwclusters = load_topic_keyword(dbfile)
purity = compare_word_cluster(coms, rwclusters)
print purity
def get_graph(filename='/home/cs/src/semantic_community/data/sample_test.db'):
from CommunityBuilder import CommunityBuilder
cb = CommunityBuilder(filename)
graph = cb.load_title_wordnet()
return graph
def get_graph(filename='/home/cs/src/semantic_community/data/sample_test.db'):
from CommunityBuilder import CommunityBuilder
cb = CommunityBuilder(filename)
graph = cb.load_title_wordnet()
return graph
