File size: 950 Bytes
e8ac98a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import asyncio
import argparse
from training.main import main as training
def main() -> None:
parser = argparse.ArgumentParser(description = "Tebakaja Model Trainer")
parser.add_argument('-a', '--algorithm', type = str, required = True,
help = 'select the algorithm to be trained (LSTM, GRU, LSTM_GRU)')
parser.add_argument('-e', '--epochs', type = int, required = True, help = 'epochs')
parser.add_argument('-b', '--batchs', type = int, required = True, help = 'batch length')
parser.add_argument('-s', '--sequences', type = int, required = True, help = 'sequences length')
args = parser.parse_args()
event_loop = asyncio.get_event_loop()
event_loop.run_until_complete(
training(
epochs = args.epochs,
batch_size = args.batchs,
algorithm = args.algorithm,
sequence_length = args.sequences
)
)
if __name__ == "__main__": main()
|