Ejemplo n.º 1
0
            #print 'Initializing joint model'
            parser = oldslavdep.OldSlavDep(words, pos, rels, w2i, c2i, args)

        for epoch in range(args.epochs):
            print('\n-----------------\nStarting epoch', epoch + 1)

            if epoch % 10 == 0:
                if epoch == 0:
                    parser.trainer.restart(learning_rate=0.001)
                elif epoch == 10:
                    parser.trainer.restart(learning_rate=0.0005)
                else:
                    parser.trainer.restart(learning_rate=0.00025)

            parser.Train(args.conll_train)

            if args.conll_dev == "N/A":
                parser.Save(
                    os.path.join(args.output, os.path.basename(args.model)))

            else:
                devPredSents = parser.Predict(args.conll_dev)

                count = 0
                lasCount = 0
                uasCount = 0
                posCount = 0
                poslasCount = 0
                for idSent, devSent in enumerate(devPredSents):
                    conll_devSent = [