Procházet zdrojové kódy

Fixed keras and tf.keras import mix

Min před 5 roky
rodič
revize
810c636673
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  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):