def setup_dork_generator(self, conf_parser, work_dir):
file_processor = dork_file_processor.DorkFileProcessor(self.dorkdb)
mnemosyne_service = None
if conf_parser.has_option('dork-db', 'mnem_service'):
if conf_parser.getboolean('dork-db', 'mnem_service') == True:
mnemosyne_service = mnem_service.Mnem_Service()
return dork_page_generator.DorkPageGenerator(self.dorkdb,
file_processor,
data_dir=self.data_dir,
mnem_service_instance=mnemosyne_service)
def setup_dork_generator(self, conf_parser, work_dir):
token_pattern = conf_parser.get('dork-db', 'token_pattern')
n_clusters = conf_parser.getint('dork-db', 'n_clusters')
max_iter = conf_parser.getint('dork-db', 'max_iter')
n_init = conf_parser.getint('dork-db', 'n_init')
file_processor = dork_file_processor.DorkFileProcessor(self.dorkdb)
mnemosyne_service = None
clusterer = cluster.Cluster(token_pattern, n_clusters, max_iter, n_init, min_df=0.0)
if conf_parser.has_option('dork-db', 'mnem_service'):
if conf_parser.getboolean('dork-db', 'mnem_service') == True:
mnemosyne_service = mnem_service.Mnem_Service()
return dork_page_generator.DorkPageGenerator(self.dorkdb,
file_processor,
clusterer,
data_dir=self.data_dir,
mnem_service_instance=mnemosyne_service)