예제 #1
0
파일: main.py 프로젝트: kingulight/CWS
        default=0
    )

    args = parser.parse_args()

    # Overriding DyNet defaults
    sys.argv.insert(1,str(args.dynet_mem))
    sys.argv.insert(1,'--dynet-mem')
    sys.argv.insert(1,str(args.dynet_l2))
    sys.argv.insert(1,'--dynet-l2')
    sys.argv.insert(1,str(args.dynet_seed))
    sys.argv.insert(1,'--dynet-seed')

    if args.vocab is not None:
        from Corpus import Corpus
        fm = Corpus.load_json(args.vocab)
    elif args.train is not None:
        from  Corpus import Corpus
        fm = Corpus(args.train)
        if args.vocab_output is not None:
            fm.save_json(args.vocab_output)
            print('Wrote vocabulary file {}'.format(args.vocab_output))
            sys.exit()
    else:
        print('Must specify either --vocab_file or --traing-data.')
        print('    (Use -h or --help flag for full option list.)')
        sys.exit()

    if args.model is None:
        print('Must specify --model or (or --write-vocab) parameter.')
        print('    (Use -h or --help flag for full option list.)')