CSDN学习社区

'''
Created on 2017年7月23日

@author: weizhen
'''
#导入库
from __future__ import division,print_function,absolute_import
import tflearn
import speech_data
import tensorflow as tf
#定义参数
#learning rate是在更新权重的时候用，太高可用很快
#但是loss大，太低较准但是很慢
learning_rate=0.0001
training_iters=300000#STEPS
batch_size=64

width=20 #mfcc features
height=80 #(max) length of utterance
classes = 10  #digits

#用speech_data.mfcc_batch_generator获取语音数据并处理成批次，
#然后创建training和testing数据
batch=word_batch=speech_data.mfcc_batch_generator(batch_size)
X,Y=next(batch)
trainX,trainY=X,Y
testX,testY=X,Y #overfit for now

#4.建立模型
#speech recognition 是个many to many的问题
#所以用Recurrent NN
#通常的RNN，它的输出结果是受整个网络的影响的
#而LSTM比RNN好的地方是，它能记住并且控制影响的点，
#所以这里我们用LSTM
#每一层到底需要多少个神经元是没有规定的，太少了的话预测效果不好
#太多了会overfitting,这里普遍取128
#为了减轻过拟合的影响，我们用dropout,它可以随机地关闭一些神经元，
#这样网络就被迫选择其他路径，进而生成想对generalized模型
#接下来建立一个fully connected的层
#它可以使前一层的所有节点都连接过来，输出10类
#因为数字是0-9，激活函数用softmax,它可以把数字变换成概率
#最后用个regression层来输出唯一的类别，用adam优化器来使
#cross entropy损失达到最小

#Network building
net=tflearn.input_data([None,width,height])
net=tflearn.lstm(net,128,dropout=0.8)
net=tflearn.fully_connected(net,classes,activation='softmax')
net=tflearn.regression(net,optimizer='adam',learning_rate=learning_rate,loss='categorical_crossentropy')


#5.训练模型并预测
#然后用tflearn.DNN函数来初始化一下模型，接下来就可以训练并预测，最好再保存训练好的模型
#Traing
### add this "fix" for tensorflow version erros
col=tf.get_collection(tf.GraphKeys.TRAINABLE_VARIABLES)
for x in col:
    tf.add_to_collection(tf.GraphKeys.VARIABLES,x)

model=tflearn.DNN(net,tensorboard_verbose=0)

while 1:  #training_iters
    model.fit(trainX, trainY, n_epoch=10, validation_set=(testX,testY), show_metric=True, batch_size=batch_size)
    _y=model.predict(X)
model.save("tflearn.lstm.model")
print(_y)

下面是训练的结果

Training Step: 3097  | total loss: [1m[32m1.51596[0m[0m | time: 1.059s
[2K
| Adam | epoch: 3097 | loss: 1.51596 - acc: 0.6324 | val_loss: 0.36655 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3098  | total loss: [1m[32m1.64602[0m[0m | time: 1.050s
[2K
| Adam | epoch: 3098 | loss: 1.64602 - acc: 0.5801 | val_loss: 0.36642 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3099  | total loss: [1m[32m1.54328[0m[0m | time: 1.052s
[2K
| Adam | epoch: 3099 | loss: 1.54328 - acc: 0.6206 | val_loss: 0.36673 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3100  | total loss: [1m[32m1.65763[0m[0m | time: 1.044s
[2K
| Adam | epoch: 3100 | loss: 1.65763 - acc: 0.5741 | val_loss: 0.36645 - val_acc: 1.0000 -- iter: 64/64
--
---------------------------------
Run id: E1W1VX
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 64
Validation samples: 64
--
Training Step: 3101  | total loss: [1m[32m1.56009[0m[0m | time: 1.328s
[2K
| Adam | epoch: 3101 | loss: 1.56009 - acc: 0.6167 | val_loss: 0.36696 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3102  | total loss: [1m[32m1.68916[0m[0m | time: 1.034s
[2K
| Adam | epoch: 3102 | loss: 1.68916 - acc: 0.5660 | val_loss: 0.36689 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3103  | total loss: [1m[32m1.58796[0m[0m | time: 1.044s
[2K
| Adam | epoch: 3103 | loss: 1.58796 - acc: 0.6078 | val_loss: 0.36627 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3104  | total loss: [1m[32m1.49236[0m[0m | time: 1.055s
[2K
| Adam | epoch: 3104 | loss: 1.49236 - acc: 0.6470 | val_loss: 0.36599 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3105  | total loss: [1m[32m1.60916[0m[0m | time: 1.028s
[2K
| Adam | epoch: 3105 | loss: 1.60916 - acc: 0.5995 | val_loss: 0.36535 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3106  | total loss: [1m[32m1.51083[0m[0m | time: 1.049s
[2K
| Adam | epoch: 3106 | loss: 1.51083 - acc: 0.6396 | val_loss: 0.36534 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3107  | total loss: [1m[32m1.63413[0m[0m | time: 1.066s
[2K
| Adam | epoch: 3107 | loss: 1.63413 - acc: 0.5865 | val_loss: 0.36566 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3108  | total loss: [1m[32m1.74167[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3108 | loss: 1.74167 - acc: 0.5373 | val_loss: 0.36556 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3109  | total loss: [1m[32m1.63324[0m[0m | time: 1.051s
[2K
| Adam | epoch: 3109 | loss: 1.63324 - acc: 0.5835 | val_loss: 0.36557 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3110  | total loss: [1m[32m1.75479[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3110 | loss: 1.75479 - acc: 0.5377 | val_loss: 0.36524 - val_acc: 1.0000 -- iter: 64/64
--
---------------------------------
Run id: 93CFSE
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 64
Validation samples: 64
--
Training Step: 3111  | total loss: [1m[32m1.64290[0m[0m | time: 1.320s
[2K
| Adam | epoch: 3111 | loss: 1.64290 - acc: 0.5839 | val_loss: 0.36560 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3112  | total loss: [1m[32m1.76515[0m[0m | time: 1.029s
[2K
| Adam | epoch: 3112 | loss: 1.76515 - acc: 0.5349 | val_loss: 0.36552 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3113  | total loss: [1m[32m1.65166[0m[0m | time: 1.050s
[2K
| Adam | epoch: 3113 | loss: 1.65166 - acc: 0.5814 | val_loss: 0.36609 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3114  | total loss: [1m[32m1.76346[0m[0m | time: 1.062s
[2K
| Adam | epoch: 3114 | loss: 1.76346 - acc: 0.5342 | val_loss: 0.36636 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3115  | total loss: [1m[32m1.65255[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3115 | loss: 1.65255 - acc: 0.5808 | val_loss: 0.36636 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3116  | total loss: [1m[32m1.55663[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3116 | loss: 1.55663 - acc: 0.6227 | val_loss: 0.36689 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3117  | total loss: [1m[32m1.67928[0m[0m | time: 1.051s
[2K
| Adam | epoch: 3117 | loss: 1.67928 - acc: 0.5729 | val_loss: 0.36726 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3118  | total loss: [1m[32m1.78375[0m[0m | time: 1.043s
[2K
| Adam | epoch: 3118 | loss: 1.78375 - acc: 0.5266 | val_loss: 0.36714 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3119  | total loss: [1m[32m1.67364[0m[0m | time: 1.041s
[2K
| Adam | epoch: 3119 | loss: 1.67364 - acc: 0.5724 | val_loss: 0.36725 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3120  | total loss: [1m[32m1.79457[0m[0m | time: 1.044s
[2K
| Adam | epoch: 3120 | loss: 1.79457 - acc: 0.5276 | val_loss: 0.36694 - val_acc: 1.0000 -- iter: 64/64
--
---------------------------------
Run id: YE812Z
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 64
Validation samples: 64
--
Training Step: 3121  | total loss: [1m[32m1.68830[0m[0m | time: 1.351s
[2K
| Adam | epoch: 3121 | loss: 1.68830 - acc: 0.5686 | val_loss: 0.36691 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3122  | total loss: [1m[32m1.79857[0m[0m | time: 1.022s
[2K
| Adam | epoch: 3122 | loss: 1.79857 - acc: 0.5227 | val_loss: 0.36642 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3123  | total loss: [1m[32m1.68557[0m[0m | time: 1.071s
[2K
| Adam | epoch: 3123 | loss: 1.68557 - acc: 0.5673 | val_loss: 0.36519 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3124  | total loss: [1m[32m1.58528[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3124 | loss: 1.58528 - acc: 0.6106 | val_loss: 0.36366 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3125  | total loss: [1m[32m1.49228[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3125 | loss: 1.49228 - acc: 0.6495 | val_loss: 0.36180 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3126  | total loss: [1m[32m1.41012[0m[0m | time: 1.052s
[2K
| Adam | epoch: 3126 | loss: 1.41012 - acc: 0.6846 | val_loss: 0.36061 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3127  | total loss: [1m[32m1.55866[0m[0m | time: 1.023s
[2K
| Adam | epoch: 3127 | loss: 1.55866 - acc: 0.6286 | val_loss: 0.35908 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3128  | total loss: [1m[32m1.46943[0m[0m | time: 1.044s
[2K
| Adam | epoch: 3128 | loss: 1.46943 - acc: 0.6657 | val_loss: 0.35735 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3129  | total loss: [1m[32m1.39050[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3129 | loss: 1.39050 - acc: 0.6992 | val_loss: 0.35632 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3130  | total loss: [1m[32m1.54006[0m[0m | time: 1.043s
[2K
| Adam | epoch: 3130 | loss: 1.54006 - acc: 0.6371 | val_loss: 0.35513 - val_acc: 1.0000 -- iter: 64/64
--
---------------------------------
Run id: YGRXY5
Log directory: /tmp/tflearn_logs/
---------------------------------
Training samples: 64
Validation samples: 64
--
Training Step: 3131  | total loss: [1m[32m1.45402[0m[0m | time: 1.336s
[2K
| Adam | epoch: 3131 | loss: 1.45402 - acc: 0.6702 | val_loss: 0.35442 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3132  | total loss: [1m[32m1.59202[0m[0m | time: 1.029s
[2K
| Adam | epoch: 3132 | loss: 1.59202 - acc: 0.6110 | val_loss: 0.35325 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3133  | total loss: [1m[32m1.50035[0m[0m | time: 1.070s
[2K
| Adam | epoch: 3133 | loss: 1.50035 - acc: 0.6499 | val_loss: 0.35195 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3134  | total loss: [1m[32m1.41417[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3134 | loss: 1.41417 - acc: 0.6849 | val_loss: 0.35042 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3135  | total loss: [1m[32m1.34060[0m[0m | time: 1.037s
[2K
| Adam | epoch: 3135 | loss: 1.34060 - acc: 0.7149 | val_loss: 0.34937 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3136  | total loss: [1m[32m1.47476[0m[0m | time: 1.039s
[2K
| Adam | epoch: 3136 | loss: 1.47476 - acc: 0.6574 | val_loss: 0.34826 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3137  | total loss: [1m[32m1.38535[0m[0m | time: 1.053s
[2K
| Adam | epoch: 3137 | loss: 1.38535 - acc: 0.6917 | val_loss: 0.34739 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3138  | total loss: [1m[32m1.51673[0m[0m | time: 1.063s
[2K
| Adam | epoch: 3138 | loss: 1.51673 - acc: 0.6413 | val_loss: 0.34637 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3139  | total loss: [1m[32m1.42892[0m[0m | time: 1.042s
[2K
| Adam | epoch: 3139 | loss: 1.42892 - acc: 0.6756 | val_loss: 0.34570 - val_acc: 1.0000 -- iter: 64/64
--
Training Step: 3140  | total loss: [1m[32m1.58217[0m[0m | time: 1.052s
[2K
| Adam | epoch: 3140 | loss: 1.58217 - acc: 0.6112 | val_loss: 0.34494 - val_acc: 1.0000 -- iter: 64/64
--
---------------------------------

这里边有一个死循环，具体怎么回事我也不太清楚。

下边是可视化训练，展示训练的图像