def get_k_best_tree_summary( interactions_path, people_path, corpus_dict_path, lda_model_path, cand_trees_path, k, people_repr_template, undirected=False): summary_kws = build_default_summary_kws_from_path( interactions_path, people_path, corpus_dict_path, lda_model_path, people_repr_template, undirected=undirected ) trees = k_best_trees(pickle.load(open(cand_trees_path)), k) return summary(trees, summary_kws=summary_kws, tablefmt='orgtbl')
def get_k_best_tree_summary(interactions_path, people_path, corpus_dict_path, lda_model_path, cand_trees_path, k, people_repr_template, undirected=False): summary_kws = build_default_summary_kws_from_path(interactions_path, people_path, corpus_dict_path, lda_model_path, people_repr_template, undirected=undirected) trees = k_best_trees(pickle.load(open(cand_trees_path)), k) return summary(trees, summary_kws=summary_kws, tablefmt='orgtbl')
def main(): import argparse import cPickle as pkl from util import json_dump parser = argparse.ArgumentParser('dump vis timeline data') parser.add_argument('--cand_trees_path', required=True) parser.add_argument('--output_path', required=True) parser.add_argument('--interactions_path', required=True) parser.add_argument('--people_path', required=True) parser.add_argument('--corpus_dict_path', required=True) parser.add_argument('--lda_model_path', required=True) parser.add_argument('--people_repr_template', type=str, default="{id}") parser.add_argument('-k', type=int, default=10) parser.add_argument('--undirected', default=False, action="store_true") args = parser.parse_args() summary_kws = build_default_summary_kws_from_path( args.interactions_path, args.people_path, args.corpus_dict_path, args.lda_model_path, args.people_repr_template, undirected=args.undirected ) trees = pkl.load(open(args.cand_trees_path)) # add hashtags if there print(len(trees)) first_node = trees[0].nodes()[0] if 'hashtags' in trees[0].node[first_node]: print('add hashtags') summary_kws['hashtags'] = {} data = run(trees, args.k, summary_kws, args.undirected) json_dump(data, args.output_path)
def main(): import argparse import cPickle as pkl from util import json_dump parser = argparse.ArgumentParser('dump vis timeline data') parser.add_argument('--cand_trees_path', required=True) parser.add_argument('--output_path', required=True) parser.add_argument('--interactions_path', required=True) parser.add_argument('--people_path', required=True) parser.add_argument('--corpus_dict_path', required=True) parser.add_argument('--lda_model_path', required=True) parser.add_argument('--people_repr_template', type=str, default="{id}") parser.add_argument('-k', type=int, default=10) parser.add_argument('--undirected', default=False, action="store_true") args = parser.parse_args() summary_kws = build_default_summary_kws_from_path( args.interactions_path, args.people_path, args.corpus_dict_path, args.lda_model_path, args.people_repr_template, undirected=args.undirected) trees = pkl.load(open(args.cand_trees_path)) # add hashtags if there print(len(trees)) first_node = trees[0].nodes()[0] if 'hashtags' in trees[0].node[first_node]: print('add hashtags') summary_kws['hashtags'] = {} data = run(trees, args.k, summary_kws, args.undirected) json_dump(data, args.output_path)