python 人工神经网络函数
时间: 2023-07-21 20:40:26 浏览: 93
Python中有多个库可以用来构建人工神经网络,其中比较常用的是TensorFlow、Keras和PyTorch等。以下是一个用Keras构建简单神经网络的代码示例:
```python
from keras.models import Sequential
from keras.layers import Dense
# 创建一个序贯模型
model = Sequential()
# 添加输入层和第一个隐藏层
model.add(Dense(units=64, activation='relu', input_dim=100))
# 添加第二个隐藏层
model.add(Dense(units=64, activation='relu'))
# 添加输出层
model.add(Dense(units=10, activation='softmax'))
# 编译模型
model.compile(loss='categorical_crossentropy',
optimizer='sgd',
metrics=['accuracy'])
```
这个模型包含三个全连接层,其中输入层和第一个隐藏层有64个神经元,第二个隐藏层有10个神经元,输出层使用softmax激活函数。模型使用交叉熵作为损失函数,随机梯度下降作为优化器,评估指标为准确率。
阅读全文
相关推荐


















