瀏覽代碼

Fixed keras and tf.keras import mix

Min 5 年之前
父節點
當前提交
810c636673
共有 1 個文件被更改,包括 2 次插入2 次删除
  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):