def main():
# missing 'TBP', 'TAF1', 'BCL11A'
#my_motifs = set(['CTCF', 'POU2F2', 'BATF', 'IRF4', 'REST', 'SPI1',
# 'MYC', 'NFKB', 'PAX5', 'TATA', 'TCF12', 'YY1'])
#print sorted(my_motifs)
# choose a motif randomly
grpd_motifs = load_motifs(sys.argv[1])
for factor, motifs in sorted(grpd_motifs.items()):
motif = motifs[0]
with open("%s.motif.txt" % motif.name, "w") as ofp:
ofp.write(">" + "\n".join(motif.lines) + "\n")
def cmp_to_brute():
motif = load_motifs(sys.argv[1]).values()[0][0]
ref_energy, ddg_array = motif.build_ddg_array()
x, part_fn = est_partition_fn(ref_energy, ddg_array, NBINS)
min_cdf = 1 - (1 - part_fn.cumsum())**2
min_pdf = np.array((min_cdf[1:] - min_cdf[:-1]).tolist() + [0.0,])
x2, part_fn_brute = est_partition_fn_by_brute_force(motif, NBINS)
part_plt, = plt.plot(x, part_fn, label='Part Fn')
min_part_plt, = plt.plot(x, min_pdf, label='Min Part Fn')
brute_plt, = plt.plot(x2, part_fn_brute, label='Brute Part Fn')
plt.legend(handles=[part_plt, min_part_plt, brute_plt])
plt.show()
return
