AI - Keras 构建模型
创始人
2024-07-31 13:30:12
0

以下是一个使用Keras库构建AI模型的代码示例:

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=10, activation='softmax'))

# 编译模型
model.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=5, batch_size=32)

# 评估模型
loss_and_metrics = model.evaluate(x_test, y_test, batch_size=128)

# 使用模型进行预测
classes = model.predict(x_test, batch_size=128)

在上面的示例中,首先导入了Keras库的Sequential和Dense模块。然后,创建了一个Sequential对象作为模型的容器。

接着,通过model.add()方法添加了一个输入层和一个隐藏层。输入层的维度为100,激活函数使用了ReLU激活函数。隐藏层使用了64个神经元。

然后,使用model.add()方法添加了一个输出层。输出层使用了10个神经元,激活函数使用了softmax函数。

接下来,使用model.compile()方法编译了模型。指定了损失函数为categorical_crossentropy,优化器为sgd(随机梯度下降),衡量指标为准确率。

然后,使用model.fit()方法训练模型。指定了训练数据x_train和标签y_train,迭代5次,每次使用32个样本进行训练。

在训练完成后,使用model.evaluate()方法评估模型。指定了测试数据x_test和标签y_test,每次使用128个样本进行评估。

最后,使用model.predict()方法使用模型进行预测。指定了测试数据x_test,每次使用128个样本进行预测,并返回预测结果。

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...