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)
