# Set up arguement parser parser = argparse.ArgumentParser() parser.add_argument( "-net", "--network_architecture", help= "String - Specify what network architecture to use - Options: (BasicRNN, EmbeddingRNN, EncDecRNN, EmbedBiRncDecRNN)" ) args = parser.parse_args() if args.network_architecture == 'BasicRNN': translator = Translator(BasicRNN, 'BasicRNN', embed=False) elif args.network_architecture == 'EmbeddingRNN': translator = Translator(EmbeddingRNN, 'EmbeddingRNN') elif args.network_architecture == 'EncDecRNN': translator = Translator(EncDecRNN, 'EncDecRNN', embed=False) elif args.network_architecture == 'EmbedBiEncDecRNN': translator = Translator(EmbedBiEncDecRNN, 'EmbedBiEncDecRNN') else: translator = Translator(EmbedBiEncDecRNN, 'EmbedBiEncDecRNN') # Load English data eng_sentences = load_data('data/small_vocab_en') # Load French data fra_sentences = load_data('data/small_vocab_fr') translator.fit(eng_sentences, fra_sentences) translator.save()