机器学习之路

Keras开发概述

2022-04-06  本文已影响0人  Nefelibatas

什么是Keras?

Keras是一个高层神经网络API,Keras由纯Python编写而成并基Tensorflow、Theano后端。Keras 为支持快速实验而生,能够把你的idea迅速转换为结果,如果你有如下需求,请选择Keras:

Keras的设计原则

典型Keras开发流程

1.定义训练数据:输入张量和目标张量。
2.定义组成的网络或model,将输入映射到目标。
3.配置学习过程:选择损失函数、优化器和需要监控的指标。
4.调用model的fit()在训练数据集上迭代。

定义model有两种办法:

使用Sequential类定义的两层model。

from keras import models
from keras import layers

model = models.Sequential()
model.add(layers.Dense(32,activation='relu',input_shape=(784,)))
model.add(layers.Dense(10,activation='softmax'))

函数式API

input_tensor = layers.Input(shape=(784,))
x = layers.Dense(32,activation='relu')(input_tensor)
out_tensor = layers.Dense(10,activation='softmax')(x)
model = models.Model(inputs=input_tensor,outputs=out_tensor)

配置学习过程,选择损失函数、优化器和需要监控的指标。

from keras import optimizers
model.compile(optimizer='rmsprop',loss='mse',metrics=['accuracy'])

训练model

model.fit(input_tensor,target_tensor,batch_size=128,epoch=10)
上一篇下一篇

猜你喜欢

热点阅读