def main():
if FLAG == 'RNN':
rnn_main(1)
elif FLAG == 'FFNN':
hidden_dim = 32
number_of_epochs = 10
ffnn_main(hidden_dim=hidden_dim, number_of_epochs=number_of_epochs)
def main():
if FLAG == 'RNN':
raise NotImplementedError
rnn_main()
elif FLAG == 'FFNN':
hidden_dim = 32
number_of_epochs = 10
ffnn_main(hidden_dim=hidden_dim, number_of_epochs=number_of_epochs)
def main():
if FLAG == 'RNN':
hidden_dim = 32
#n_layers = 2
number_of_epochs = 10
rnn_main(hidden_dim, number_of_epochs)
elif FLAG == 'FFNN':
hidden_dim = 32
number_of_epochs = 10
ffnn_main(hidden_dim=hidden_dim, number_of_epochs=number_of_epochs)
def main():
if args.model == 'RNN':
rnn_main(args.name, args.embedding, args.hidden, args.layers,
args.epochs, args.RNNcore)
elif args.model == 'FFNN':
ffnn_main(args.name,
hidden_dim=args.hidden,
number_of_epochs=args.epochs,
n_layers=args.layers)
else:
print('Incompatible model declaration')
return
def main():
if FLAG == 'RNN':
#raise NotImplementedError
embedding_dim = 128
hidden_dim = 64
number_of_epochs = 10
num_layers = 2
rnn_main(embedding_dim=embedding_dim,
hidden_dim=hidden_dim,
number_of_epochs=number_of_epochs,
num_layers=num_layers)
elif FLAG == 'FFNN':
hidden_dim = 32
number_of_epochs = 10
ffnn_main(hidden_dim=hidden_dim, number_of_epochs=number_of_epochs)
def main():
if FLAG == 'RNN':
hidden_dim1 = 256
hidden_dim2 = 128
hidden_dim3 = 64
hidden_dim4 = 32
number_of_epochs = 5
rnn_main(h1=hidden_dim1,
h2=hidden_dim2,
h3=hidden_dim3,
h4=hidden_dim4,
number_of_epochs=number_of_epochs)
elif FLAG == 'FFNN':
hidden_dim = 32
number_of_epochs = 10
ffnn_main(hidden_dim=hidden_dim, number_of_epochs=number_of_epochs)
def main():
if FLAG == 'RNN':
hidden_dim = 32
n_layers = 3
number_of_epochs = 10
rnn_main(hidden_dim=hidden_dim, n_layers, number_of_epochs=number_of_epochs)
def main():
if FLAG == 'RNN':
rnn_main(hidden_dim=32, batch_size=1)