changed imports

models/end_to_end.py

@@ -7,7 +7,7 @@ from sklearn.metrics import accuracy_score
 from sklearn.preprocessing import OneHotEncoder
 from tensorflow.keras import layers, losses
 from tensorflow.keras import backend as K
-from models.custom_layers import ExtractCentralMessage, OpticalChannel, DigitizationLayer, BitsToSymbols, SymbolsToBits
+from custom_layers import ExtractCentralMessage, OpticalChannel, DigitizationLayer, BitsToSymbols, SymbolsToBits
 import itertools
 
 

models/new_model.py

@@ -1,8 +1,8 @@
 import tensorflow as tf
 from tensorflow.keras import layers, losses
-from models.custom_layers import ExtractCentralMessage, OpticalChannel
-from models.end_to_end import EndToEndAutoencoder
-from models.custom_layers import BitsToSymbols, SymbolsToBits
+from custom_layers import ExtractCentralMessage, OpticalChannel
+from end_to_end import EndToEndAutoencoder
+from custom_layers import BitsToSymbols, SymbolsToBits
 import numpy as np
 import math