From 6be1268feccfd21f4482a5957f72221347588893 Mon Sep 17 00:00:00 2001 From: Ludovic Moncla <moncla.ludovic@gmail.com> Date: Fri, 25 Nov 2022 14:39:08 +0100 Subject: [PATCH] Update Predict_LGE.py --- scripts/Predict_LGE.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/Predict_LGE.py b/scripts/Predict_LGE.py index fabc4cf..7d474e1 100644 --- a/scripts/Predict_LGE.py +++ b/scripts/Predict_LGE.py @@ -186,7 +186,7 @@ if __name__ == '__main__': pred = predict(model, data_loader, device) encoder_filename = "models/label_encoder.pkl" - with open(path+encoder_filename, 'rb') as file: + with open(path + encoder_filename, 'rb') as file: encoder = pickle.load(file) p2 = list(encoder.inverse_transform(pred)) -- GitLab