数据集和代码均已上传到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"""
f