> 数据挖掘与机器学习 > 深度学习之Keras入门

深度学习之Keras入门

数据挖掘与机器学习 fireling 8469℃

Keras是一个上层的神经网络API，它由纯Python编写而成并基Tensorflow、CNTK或者Theano为后端，其项目地址位于https://github.com/fchollet/keras 。

Keras的核心数据结构是“模型”，这是一种组织网络层的方式。最简单的模型是Sequential模型，这是一系列网络层按顺序构成的栈。

Keras环境搭建

Keras依赖下面几个库：


numpy, scipy
yaml
HDF5 and h5py (Optional)
cuDNN (Optional)

具体搭建步骤：

  • 安装开发包

sudo apt install -y python-dev python-pip python-nose gcc g++ git gfortran vim
  • 安装运算加速库

sudo apt install -y libopenblas-dev liblapack-dev libatlas-base-dev
  • 安装keras

pip install keras
  • 安装tensorflow(使用TensorFlow为后端)

pip install tensorflow # for Python 2.7
pip install tensorflow-gpu # for Python 2.7 and GPU
  • keras中mnist数据集测试

git clone https://github.com/fchollet/keras.git
cd keras/examples/
python mnist_mlp.py

Keras上层API使用

搭建模型

搭建一个Sequential模型：


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

可以通过向Sequential模型传递一个layer的list来构造该模型：


model = Sequential([
    Dense(units=64, input_shape=(100,)),
    Activation('relu'),
    Dense(units=10),
    Activation('softmax'),
])

也可以通过.add()方法一个个的将layer加入Sequential模型中：


model = Sequential()
model.add(Dense(units=64, input_dim=100))
model.add(Activation('relu'))
model.add(Dense(units=10))
model.add(Activation('softmax'))

编译模型

这一步用来配置模型的学习流程，编译模型时必须指明损失函数、优化器、度量：


model.compile(loss='categorical_crossentropy', optimizer='sgd', metrics=['accuracy'])
  • 损失函数

该参数为模型试图最小化的目标函数，它可为预定义的损失函数名，如categorical_crossentropy、mse，也可以为一个损失函数。
  • 优化器

该参数可指定为已预定义的优化器名，如rmsprop、adagrad，或者为一个Optimizer类的对象。
  • 度量

对分类问题，一般将该列表设置为metrics=['accuracy']。指标可以是一个预定义指标的名字，也可以是一个用户定制的函数。

如果你需要的话，你可以自己定制损失函数：


from keras.optimizers import SGD
model.compile(loss='categorical_crossentropy', optimizer=SGD(lr=0.01, momentum=0.9, nesterov=True))

训练模型

我们在训练数据上按batch进行一定次数的迭代来训练网络：


model.fit(x_train, y_train, epochs=5, batch_size=32)

我们也可以手动将一个个batch的数据送入网络中训练：


model.train_on_batch(x_batch, y_batch)

评估模型


loss_and_metrics = model.evaluate(x_test, y_test, batch_size=128)

模型预测


classes = model.predict(x_test, batch_size=128)

Keras代码举例

以“基于多层感知器(Multilayer Perceptron, MLP)的sigmoid二分类”为例，代码如下：


import keras
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation

# Generate dummy data, with 2 classes (binary classification)
import numpy as np
x_train = np.random.random((1000, 20))
y_train = np.random.randint(2, size=(1000, 1))
x_test = np.random.random((100, 20))
y_test = np.random.randint(2, size=(100, 1))

model = Sequential()
model.add(Dense(64, activation='relu', input_dim=20))
model.add(Dropout(0.5))
model.add(Dense(64, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(1, activation='sigmoid'))

model.compile(loss='binary_crossentropy',
              optimizer='sgd',
              metrics=['accuracy'])

# Train the model, iterating on the data in batches of 128 samples
model.fit(x_train, y_train,
          epochs=20,
          batch_size=128)

score = model.evaluate(x_test, y_test, batch_size=128)
print(score)

运行结果

keras_test

转载请注明：宁哥的小站 » 深度学习之Keras入门

喜欢 (7)