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