Skip to content

Model

Bao, Ziyuan L requested to merge model into master

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)

Merge request reports

Loading