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