Przeglądaj źródła

Fixed keras and tf.keras import mix

Min 4 lat temu
rodzic
commit
810c636673
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      models/end_to_end.py

+ 2 - 2
models/end_to_end.py

@@ -1,12 +1,12 @@
 import math
 import math
 
 
-import keras
+from tensorflow import keras
 import tensorflow as tf
 import tensorflow as tf
 import numpy as np
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.pyplot as plt
 from matplotlib import collections as matcoll
 from matplotlib import collections as matcoll
 from sklearn.preprocessing import OneHotEncoder
 from sklearn.preprocessing import OneHotEncoder
-from keras import layers, losses
+from tensorflow.keras import layers, losses
 
 
 
 
 class ExtractCentralMessage(layers.Layer):
 class ExtractCentralMessage(layers.Layer):