keras写一个5分类的代码
时间: 2023-05-17 07:07:05 浏览: 54
以下是一个使用 Keras 实现 5 分类的代码示例:
```python
import numpy as np
from keras.models import Sequential
from keras.layers import Dense, Activation
# 生成随机数据
data = np.random.random((1000, 10))
labels = np.random.randint(5, size=(1000, 1))
# 将标签转换为 one-hot 编码
one_hot_labels = keras.utils.to_categorical(labels, num_classes=5)
# 定义模型
model = Sequential()
model.add(Dense(32, activation='relu', input_dim=10))
model.add(Dense(5, activation='softmax'))
# 编译模型
model.compile(optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=['accuracy'])
# 训练模型
model.fit(data, one_hot_labels, epochs=10, batch_size=32)
```
这个模型使用了一个包含 32 个神经元的隐藏层和一个包含 5 个神经元的输出层。激活函数使用了 relu 和 softmax。训练数据包含 1000 个样本,每个样本有 10 个特征。模型使用了 rmsprop 优化器和交叉熵损失函数进行训练。