Skip to content
Snippets Groups Projects
Commit 6be1268f authored by Ludovic Moncla's avatar Ludovic Moncla
Browse files

Update Predict_LGE.py

parent dfede5ac
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
scripts/Predict_LGE.py
+ 1
− 1
View file @ 6be1268f
...@@ -186,7 +186,7 @@ if __name__ == '__main__': ...@@ -186,7 +186,7 @@ if __name__ == '__main__':
pred = predict(model, data_loader, device) pred = predict(model, data_loader, device)
encoder_filename = "models/label_encoder.pkl" 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) encoder = pickle.load(file)
p2 = list(encoder.inverse_transform(pred)) p2 = list(encoder.inverse_transform(pred))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment