TensorFlow.js 入门并动手编写娱乐版的双色球在线预测
背景
一年前就知道了 TensorFlow js 版的存在,一直没有去玩一下。最近一次偶然的机会,要协助同事处理过一个 TensorFlow.js
相关的 Demo 小程序,要使用 TensorFlow.js
结合小程序的照相机 API 功能,实现**微信小程序扫描识别物体功能**。该 Demo 主要是由 TensorFlow.js 官方项目 – https://github.com/tensorflow/tfjs-wechat/tree/master/demo/mobilenet 改造而来。后面开始看官方 demo 开始由最简单的 《TensorFlow.js 实现线性回归》 的例子开始学习,并捣鼓出这个《TensorFlow.js 娱乐版的双色球在线预测》。
工具由来
为什么叫《TensorFlow.js 娱乐版的双色球在线预测》?哈哈,因为双色球开的期数总数才 2510+ 期,作为训练模型 model
学习的基础数据也太小了。还有本身电脑的计算能力也有限,如果硬把学习次数 epochs
加非常高,可能电脑要运行几天才可以学习完并预测出结果哈,大家可以试试选用期数为 00000-20000
来作为训练模型 model
然后把学习次数 epochs
加高来进行预测。
费话不多说,直接关键源码分析
$(document).ready(function() {
runTf('00000-20000', 1, 29999)
// 默认打开运行预测 训练模型期号数据是 00000-20000,学习次数是 1 次,预测的期号是 29999
$('#btn').click(function` `() { // 点击按钮绑定处理
const modelVal = $('#model').val()
const epochsVal = $('#epochs').val()
const roundVal = $('#round').val()
const tips = '<span class="tips pulse animated infinite">正在预测...</span>';
if (!modelVal) {
alert('请先选择历史数据模型训练(model)')
return false
}
if (!epochsVal) {
alert('请先输入学习次数(epochs)')
return false
}
if (!roundVal) {
alert('请先输入待预测的期号')
return false
}
$('#micro-out-div').html(tips)
runTf(modelVal, Number(epochsVal), Number(roundVal))
})
function runTf(historyModel, epochs, round) { // 关键函数
$('#table').load(`./ssq/history-${historyModel}.htm`, function` `() {
// 加载相应的训练模型数据,来源于 500 彩票 http://kaijiang.500.com
const rawArr = []
const xsArr = []
const yxArr = []
// 通过 dom 操作把源数据由 table 中抽取出来
$('#table .t_tr1').each(function` `() {
let This = $(this);
let str = ''
This.find('td').each(function` `(n) {
if` `(n === 0) {
str += $(this).text()
} else` `if` `(n < 8 && n > 0) {
str += '|'` `+ $(this).text()
}
})
let arr = str.split('|')
arr = arr.map((_) => Number(_))
rawArr.push(arr)
xsArr.push(arr[0])
yxArr.push(arr.slice(1))
})
async function` `run() { // 关键代码
// Sequential 序贯模型,序贯模型是函数式模型的简略版,为最简单的线性、从头到尾的结构顺序,不分叉,是多个网络层的线性堆叠。
const model = tf.sequential()
// 使用.add() 方法将各层添加到模型中
model.add(tf.layers.dense({ units: 7, inputShape: [1] })) // Dense,支持通过参数 inputShape 指定输入尺寸,units: 该层的神经单元结点数(输出结果是 7 个球)
// compile({loss, optimizer, metrics}) 编译模型,定义损失函数,优化函数,绩效评估函数
model.compile({ loss: 'binaryCrossentropy', optimizer: 'sgd'` `}) // metrics 可参考:https://js.tensorflow.org/api/1.7.2/#Metrics
console.log('开始,gogogo!')
// 下面其实定义使用 tensor / tensor1d / tensor2d 都没关系,最终 tensor1d / tensor2d ...3d... 都跟 tensor 效果一样,语法糖来的
const xs = tf.tensor(xsArr) // 定义一个张量(这里是历史期号)
console.log(xs.shape) // 打印输入尺寸
const ys = tf.tensor(yxArr) // 定义一个张量(这里是上面 xs 的历史期号对应所有的开奖结果)
console.log(ys.shape) // 打印输入尺寸
console.time('训练时间')
console.log('开始训练...')
await model.fit(xs, ys, { epochs: epochs }) // 导入数据进行训练 可参考:https://js.tensorflow.org/api/1.7.2/#tf.LayersModel.fit
console.timeEnd('训练时间')
console.log('结束训练')
console.time('预测时间')
console.log('开始预测...')
const predicted = model // 为什么叫娱乐版的呢?因为下面出来的结果太失望了,要自己重新处理
.predict(tf.tensor2d([round], [1, 1])) // 预测自己输入的期号为 round 的结果
.abs() // 取绝对值
.floor() // 去掉小数点
.dataSync() // 得到的是一个 Float32Array,如果学习次数 epochs 太少的话,结果类似 [20661, 28071, 7391, 10978, 7883, 9646, 5441],WTF!太差了吗?
.map((_, n) => { // 自己按号码规则求余优化成最终结果
return` `n !== 6 ? _ % 33 || 33 : _ % 16 || 16
})
const obj = {} // 判断出来的红球有没有重复
predicted.forEach((_, n) => {
if` `(n !== 6) {
obj[_] = 1
}
})
if` `(Object.keys(obj).length < 6) { // 什么?出来的红球有重复?帮我重新预测吧
run()
return` `false
}
// 下面是 dom 操作,打印预测结果
const predictedHtml = [...predicted] .map((_, n) => {
return` `n !== 6 ? `<i>${_}</i>` : `<b>${_}</b>
})
.join(' | ')
$('#micro-out-div').html(
'期号 '` `+ round + ' 预测结果为:'` `+ predictedHtml
)
console.timeEnd('预测时间')
console.log('结束预测')
}
run()
})
}
})
可能会遇到的问题
Error when checking input: expected dense_Dense1_input to have x dimension(s). but got array with shape y,z
或者 Error when checking input: expected dense_Dense1_input to have 3 dimension(s). but got array with shape
或者 Error when checking target: expected dense_Dense1 to have shape [,1], but got array with shape [6,7]
解决方式:一般是需要相应修改代码 model.add(tf.layers.dense({ units: 7, inputShape: [1] }))
中的 units 与 inputShape 的值就可以了
感慨良多
回忆起 10 几年前自己曾经在图书馆看过一本书名叫《神经网络与卷积学习》还是啥的,隐约回忆起类似下面一个画面。
TensorFlow.js 机器学习
还想起了当时书中介绍了“权重”、“权值”等概念。人工智能的概念与实现理论基础几十年前已经有了,受限与计算机的算力,一直没有发展起来,到最近十几年半导体行业与计算机计算能力的快速发展,人工智能才逐渐发展起来。
而自己机缘巧合,进入和开启了前端的职业生涯,在图形图像处理方向的学习,需要作图要用上向量与矩阵计算,需要重新拾起大学、高中、小学的数学知识;还有一段时间沉迷于对一些经典算法的练脑、学习与进步;感觉自己对未来的 AI 世界充满憧憬与期待。
参考引用
https://js.tensorflow.org/api/1.7.2/#Tensors https://github.com/tensorflow/tfjs-examples/tree/master/getting-started https://blog.csdn.net/qq_38806886/article/details/83892671 https://www.cnblogs.com/wj-1314/p/9579490.html
本文作者:Nelson Kuang,欢迎大家留言及多多指教
版权声明:欢迎转载学习 => 请标注信息来源于http://www.a4z.cn/fe/2020/05/29/tensorflow-js/