5. 训练模型

---

数据集和代码均已上传到Github中，欢迎大家下载使用。

Github地址：https://github.com/JasonZhang156/Sound-Recognition-Tutorial

如果这个教程对您有所帮助，请不吝贡献您的小星星Q^Q.

---

训练模型

我们采用交叉验证（5-fold cross validation）的方式训练模型，即预先将数据集划分为5个不重叠的子集，每次采用其中的4个子集进行训练，另一个子集做测试， 这样重复5次，使得每个子集都作为一个测试集进行训练。这种方式能够很好的评测我们模型泛化性能的好坏。子集的划分在”特征提取“部分已经介绍过了，这里就不过多赘述。关于模型训练的细节直接看代码：

# -*- coding: utf-8 -*-
"""
@author: Jason Zhang
@github: https://github.com/JasonZhang156/Sound-Recognition-Tutorial
"""

from keras.callbacks import TensorBoard, ModelCheckpoint, LearningRateScheduler
import tensorflow as tf
import keras.backend as K
import esc10_input
import numpy as np
import models
import os


def use_gpu():
    """Configuration for GPU"""
    from keras.backend.tensorflow_backend import set_session
    os.environ['CUDA_VISIBLE_DEVICES'] = str(0)   # 使用第一台GPU
    config = tf.ConfigProto