예제 #1
0
fname = "".join([str(n),"_Mchains.p"])
Mchains = pickle.load(open(fname,"r"))

print "fitting model to test set..."
a = 0
for line in fileinput.input():
    l = line.strip("\n")
    l = l.split("\t")
    l[0] = l[0].split("_")[0]
    if len(l[1])<=n:
        continue
    target = l[0]
    
    answer = np.array(Mchains.keys())
    
    estimates = np.zeros( (len(Mchains.keys()),1))
    
    temp = create_chain(len(ix),1)
    l[1] = l[1].replace("N",randomN())
    temp = process_read(l[1],temp,n,ix)
    temp = temp/float(sum(temp))
    for k,val in enumerate(Mchains.keys()):
        prob = euclid(temp,Mchains[val])
        estimates[k] = np.log2(prob)
    best = np.argmin(estimates)
    print answer[best], target
    if answer[best] == target:
        a+=1
    
print "total of ",a, "fitted properly"
예제 #2
0
n=5
ix = create_index(n)

'''create a dictionnary containing all markov chains (order 6) for each chromosome'''
Mchains = {}
probchr = {}
a = 0
for line in fileinput.input():
    l = line.strip("\n")
    l = l.split("\t")
    l[0] = l[0].split("_")[0]
    if len(l[1])<=n:
        continue
    if not ( l[0] in Mchains.keys()):
        Mchains[l[0]] = create_chain(len(ix),1)
        probchr[l[0]] = 0
    temp = create_chain(len(ix),1)
    l[1] = l[1].replace("N",randomN())
    Mchains[l[0]]+=process_read(l[1],temp,n,ix)
    probchr[l[0]]+=1
sum_prob = float(np.sum(probchr.values()))

#calculating Poisson distribution parameters (mu) for each Markov chain
for k in Mchains.keys():
    Mchains[k] = Mchains[k]/float(sum(Mchains[k]))
    probchr[k] = probchr[k]/sum_prob
    


''' For each chromosome, the occurences are converted to