Esempio n. 1
0
    def setUp(self):

        fname = 'friends.txt'
        f = open(fname, 'r')
        self.txt = f.read()
        f.close()


        c = Corpus(self.txt)
        c.brownInit(500)
        rnn = RNN(100, c.V, 50)
        rnn.load('rnn.save')

        self.trainer = Trainer(c,rnn, nepochs=50, alpha = 1.8)
Esempio n. 2
0
from rnn import *
from dataLoader import Corpus
from trainer import Trainer

fname = 'shakespear.txt'
f = open(fname, 'r')

txt = f.read()

f.close()

c = Corpus(txt)
c.brownInit(10000)

rnn = RNN(100, c.V, 100)
# rnn = RNN.load('rnn.save')
rnn.load('rnn.save')

# rnn = RNN(100, c.V, 50)

trainer = Trainer(c, rnn, nepochs=50, alpha=0.9)
trainer.generate_sequence()
# trainer.train()
trainer.mainEventLoop()
Esempio n. 3
0
    def test_perplexicity(self):

        c = Corpus("asdada asdaa asd adada dadada. asdas dasd.da ad.a d.sa da asd")
        c.brownInit(500,0)
        p = self.trainer.calcPerplexicity(c)
        print Fore.CYAN, p, "\n"
Esempio n. 4
0
from rnn import *
from dataLoader import Corpus
from trainer import Trainer

fname = 'shakespear.txt'
f = open(fname, 'r')

txt = f.read()

f.close()

c = Corpus(txt)
c.brownInit(10000)

rnn = RNN(100, c.V, 100)
# rnn = RNN.load('rnn.save')
rnn.load('rnn.save')

# rnn = RNN(100, c.V, 50)

trainer = Trainer(c,rnn, nepochs=50, alpha = 0.9)
trainer.generate_sequence()
# trainer.train()
trainer.mainEventLoop()