def do_reset(self):
'''reset PMI matrix by deleting all related collections'''
for c in (self._F_all, self._F_i, self._F_p, self._F_ip,
self._pmi_ip, self._max_pmi_ip):
fullname = mongodb.fullname(self.db[c])
print >>sys.stderr, 'resetting %s ...' % fullname
self.db.drop_collection(self.db[c])
print >>sys.stderr, 'resetting %s: done.' % fullname
def __init__(self, db, matrix, batch=100):
'''initializes class with information necessary for calculating PMI scores'''
self.db = db
self.matrix = matrix
self.fullname = mongodb.fullname(self.db[self.matrix])
self.batch = batch
self.argv = self.get_args()
self.argc = len(self.argv)
self._F_all = '%s_F_all' % self.matrix
self._F_i = '%s_F_i' % self.matrix
self._F_p = '%s_F_p' % self.matrix
self._F_ip = '%s_F_ip' % self.matrix
self._pmi_ip = '%s_pmi_ip' % self.matrix
self._max_pmi_ip = '%s_max_pmi_ip' % self.matrix
self.F_all = self.get_F_all()
def reset_matrix(db, matrix):
for c in get_matrix_collections(db, matrix):
fullname = mongodb.fullname(db[c])
print >> sys.stderr, 'resetting %s ...' % fullname
db.drop_collection(c)
print >> sys.stderr, 'resetting %s: done' % fullname
def reset_matrix(db, matrix):
for c in get_matrix_collections(db, matrix):
fullname = mongodb.fullname(db[c])
print >>sys.stderr, 'resetting %s ...' % fullname
db.drop_collection(c)
print >>sys.stderr, 'resetting %s: done' % fullname
