Model
moved model utils into model.py updated model unittest
Use following code in main to get has_aki value (0 or 1 integer)
model = load_model('../lstm_model.pth')
has_aki = inference(model, preprocessed_input, device)
moved model utils into model.py updated model unittest
Use following code in main to get has_aki value (0 or 1 integer)
model = load_model('../lstm_model.pth')
has_aki = inference(model, preprocessed_input, device)