基于tensorflow keras搭建模块化神经网络示例
时间: 2023-07-08 16:54:34 浏览: 61
好的,以下是一个基于TensorFlow Keras搭建模块化神经网络的示例代码:
```python
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Activation
# 定义模型
def build_model(input_size, hidden_layers, num_classes):
model = Sequential()
model.add(Dense(hidden_layers[0], input_dim=input_size))
model.add(Activation('relu'))
for i in range(1, len(hidden_layers)):
model.add(Dense(hidden_layers[i]))
model.add(Activation('relu'))
model.add(Dense(num_classes))
return model
# 测试
input_size = 784
hidden_layers = [256, 128]
num_classes = 10
model = build_model(input_size, hidden_layers, num_classes)
model.summary()
```
这个示例代码同样定义了一个三层神经网络,包括一个输入层、两个隐层和一个输出层。其中,使用了TensorFlow Keras中的Sequential模型,通过add方法添加了各个层。同时,这个代码还支持模型的构建和测试。通过调用model.summary()方法,可以查看模型的结构和参数数量。