本篇继续上篇的未完部分。前面介绍完了使用单个数据训练模型的例子,本篇继续介绍使用批量数据训练模型的例子。
2. 批量数据的训练
(1)创建项目并导入SDK
按照上一篇文章的介绍进行项目创建,并导入SDK,详细内容请参考上一篇。
(2)编码
整体代码分为四部分,即神经网络初始化、训练数据读取、网络模型训练、模型保存,其中的数据读取和模型训练两部分放在循环之中,以便读取一批的数据进行训练。整体的代码如下所示:
void main(void)
{
FILE* pFile = NULL;
char strFileName[256] = { 0 };
int nPieces = 0; //训练数据中(一局对弈)的棋子数
int narrSteps[1000] = { 0 }; //对弈中从第一子到最后一子的信息:X坐标、Y坐标、落子者(黑为-1,白为1)
int nFileIndexStart = 1; //第一个数据文件的名称所对应的数字
int* parrSteps = NULL;
InitWithoutModelFile(15, 15, 5);
sprintf(strFileName, "D:/Data/%d.txt", nFileIndexStart++);
while ((pFile = fopen(strFileName, "r")) != NULL)
{
nPieces = 0;
parrSteps = narrSteps;
while (fscanf(pFile, "%d %d %d", parrSteps, parrSteps + 1, parrSteps + 2) == 3)
{
parrSteps += 3;
nPieces++;
}
fclose(pFile);
TrainNetwork(narrSteps, nPieces, 15, 15);
sprintf(strFileName, "D:/Data/%d.txt", nFileIndexStart++);
}
SaveModel("D:/Model/model.mod");
}
首先,调用SDK的函数InitWithoutModelFile(15,15,5)进行神经网络初始化,此处表示从空模型初始化为15x15棋盘上的五子棋,若想在已有模型上进行追加训练,则调用的是InitFromModelFile()函数。若想训练的是六连子模型,则只需将第三个参数从5改为6即可。
然后,使用一个while循环来逐个打开和读取数据文件,并使用读取的数据调用TrainNetwork()函数来训练神经网络模型。为了便于循环读取批量的数据文件,需要将数据文件的名称按计数的方式进行命名,如下图所示:
此处的代码将从nFileIndexStart的值表示的那个文件开始依次读取文件,直到文件读取失败的时候退出循环,因而要保证所有的数据文件的名称对应的数字是连续的,若中间有一个数字没找到,则后续比该数字大的文件均不会读取。例如:若10.txt被删除掉了,则此处只会读去1.txt到9.txt文件进行训练,11.txt开始往后的都不会读取,非数字命名的文件也不会读取。
虽然本例是使用一批的数据来训练神经网络,但是,每读出一个数据文件就要调用一次训练函数TrainNetwork(),因而该函数的调用是放在循环内的。
最后,在模型训练结束之后调用SDK函数SaveModel()保存训练得到的模型。
(3)测试
至此为止,批量数据训练的例子已经编码完成,运行程序之后可以在D盘的Model目录下看到新训练出的模型文件model.mod,可以将该模型用于之前文章中介绍的“神经网络五子棋(家庭版)”等软件中进行测试。
本例的运行时间可能较长,尤其是数据量达到上万条之后,可能需要等待几个小时才能运行结束并生成模型文件,这可能导致等待中的茫然。为了让训练的人能够时刻把握进度,可以在循环中打印出当前正在处理的数据文件名称:
printf("%s\n", strFileName);
结果如下图所示:
(4)后续
批量数据训练网络模型的例程已经介绍完毕,其它的例程请继续关注后续文章。
本例的完整代码如下所示:
#include "stdio.h"
#include "Inter.h"
#pragma comment(lib, "AIWZQDll.lib")
void main(void)
{
FILE* pFile = NULL;
char strFileName[256] = { 0 };
int nPieces = 0; //训练数据中(一局对弈)的棋子数
int narrSteps[1000] = { 0 }; //对弈中从第一子到最后一子的信息:X坐标、Y坐标、落子者(黑为-1,白为1)
int nFileIndexStart = 1; //第一个数据文件的名称所对应的数字
int* parrSteps = NULL;
InitWithoutModelFile(15, 15, 5);
sprintf(strFileName, "D:/Data/%d.txt", nFileIndexStart++);
while ((pFile = fopen(strFileName, "r")) != NULL)
{
printf("%s\n", strFileName);
nPieces = 0;
parrSteps = narrSteps;
while (fscanf(pFile, "%d %d %d", parrSteps, parrSteps + 1, parrSteps + 2) == 3)
{
parrSteps += 3;
nPieces++;
}
fclose(pFile);
TrainNetwork(narrSteps, nPieces, 15, 15);
sprintf(strFileName, "D:/Data/%d.txt", nFileIndexStart++);
}
SaveModel("D:/Model/model.mod");
}