keras-gpu test code

最新推荐文章于 2023-11-17 12:23:34 发布

_iamv

最新推荐文章于 2023-11-17 12:23:34 发布

阅读量385

点赞数 2

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/vera425/article/details/96835374

本文介绍使用Keras库创建一个包含两层的神经网络模型。首先导入必要的模块，然后通过Sequential类建立模型，并添加两个全连接层(Dense)，分别设置激活函数为ReLU和Sigmoid。最后加载随机生成的数据并进行预测。

import keras
from keras.models import Sequential
from keras.layers import Dense

model = Sequential()
model.add(Dense(32, activation='relu',input_dim=100))
model.add(Dense(2, activation='sigmoid'))
model.summary()

import numpy as np
data = np.random.random((1000, 100))
model.predict(data)