首页 > 解决方案 > “密集”对象没有属性“操作”

问题描述

我正在尝试使用 tensorflow.keras 制作一个完全连接的模型,这是我的代码

from tensorflow.keras.models import Model
from tensorflow.keras.layers import Input, Dense, Flatten

def load_model(input_shape):
  input = Input(shape = input_shape)
  dense_shape = input_shape[0]
  x = Flatten()(input)
  x = Dense(dense_shape, activation='relu')(x)
  x = Dense(dense_shape, activation='relu')(x)
  x = Dense(dense_shape, activation='relu')(x)
  x = Dense(dense_shape, activation='relu')(x)
  x = Dense(dense_shape, activation='relu')(x)
  output = Dense(10, activation='softmax')

  model  = Model(input , output)
  model.summary()
  return model

但是当我调用模型时

model = load_model((120,))

我有这个错误

'Dense' object has no attribute 'op'

我怎样才能解决这个问题?

标签: pythontensorflowmachine-learningkerasdeep-learning

解决方案


(x)在输出层之后丢失了。尝试

output = Dense(10 , activation = 'softmax')(x)

推荐阅读