diff --git a/scripts/Predict_LGE.py b/scripts/Predict_LGE.py index fabc4cfe10abf397255f9bea0d4c5c6ce9f85a75..7d474e1fa064f29045b74ccb9a981413787bd711 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))