python - 如何在 Tensorflow 中从 tf.keras 导入 keras？

Question

import tensorflow as tf
import tensorflow 

from tensorflow import keras
from keras.layers import Dense

我收到以下错误

from keras.layers import Input, Dense
Traceback (most recent call last):

  File "<ipython-input-6-b5da44e251a5>", line 1, in <module>
    from keras.layers import Input, Dense

ModuleNotFoundError: No module named 'keras'

我该如何解决这个问题？

注意:我使用的是 Tensorflow 1.4 版

Answer 1

像这样使用 tensorflow 中的 keras 模块:

将 tensorflow 导入为 tf

导入类

from tensorflow.python.keras.layers import Input, Dense

或直接使用

dense = tf.keras.layers.Dense(...)

编辑 TensorFlow 2

from tensorflow.keras.layers import Input, Dense

其余的保持不变。