def generate_all_features(self):
all_tokens = set()
for directory in self.tokenized_input_dirs:
print "Generating for dir", directory
for f in get_files(directory):
tokens = set([token for token in re.split(' |\r\n', open(f).read())])
all_tokens = all_tokens | tokens
pickle.dump(list(all_tokens), open('token_list.pkl', 'w'))
print 'Dumped features'
def generate_popular_features(self):
all_tokens = set()
for directory in self.tokenized_input_dirs:
print "Generating for directory", directory
intermediate_tokens = set()
for f in get_files(directory):
dir_pop = defaultdict(lambda: 0)
tokens = [token for token in re.split(' |\r\n', open(f).read())]
for token in tokens:
dir_pop[token] += 1
ordered = sorted(dir_pop.items(), key=lambda x: x[1], reverse=True)
if len(ordered) > 50:
ordered = ordered[:50]
ordered = set([x[0] for x in ordered])
intermediate_tokens = intermediate_tokens | ordered
intermediate_tokens = sorted(list(intermediate_tokens), key=lambda x: dir_pop[x], reverse=True)
if len(intermediate_tokens) > 10000:
intermediate_tokens = intermediate_tokens[:10000]
intermediate_tokens = set(intermediate_tokens)
all_tokens = (all_tokens - intermediate_tokens) | (intermediate_tokens - all_tokens)
pickle.dump(list(all_tokens), open('token_list.pkl', 'w'))
