在这篇文章中,我们将利用 tensorflow.js,d3.js 和 web 的力量使训练模型的过程可视化,以预测棒球数据中的坏球(蓝色区域)和好球(橙色区域)。 随着我们的进展,我们将模型在整个训练过程中理解的打击区域可视化。您可以通过访问此 observable 笔记本在浏览器中运行此模型。
注:observable链接
https://beta.observablehq.com/@nkreeger/visualizing-ml-training-using-tensorflow-js-and-baseball-d
如果你不熟悉棒球的击球区,这里有一篇详细的文章。
上面的 gif 可视化神经网络学习调用坏球(蓝色区域)和好球(橙色区域)在每个训练步骤之后,热图会根据模型的预测进行更新
使用 observable 直接在浏览器中运行此模型。
注:文章链接
https://beta.observablehq.com/@nkreeger/visualizing-ml-training-using-tensorflow-js-and-baseball-d
体育运动中的高级指标
当今的职业体育环境中充斥着大量的数据。这些数据被团队,业余爱好者和粉丝应用于各种用例中。感谢像 tensorflow 这样的框架 - 这些数据集已准备好应用于机器学习。
美国职业棒球大联盟先进媒体(mlbam)的 pitchf/x
美国职业棒球大联盟先进媒体(mlbam)发布了一个可供公众研究的大型数据集。该数据集包含有关过去几年在美国职业棒球大联盟比赛中投掷的投球的传感器信息。 利用这个数据集,我们已编写了一个包含 5,000 个样本的训练集(2,500 个坏球和 2,500 个好球)。
以下是训练数据中前几个字段的示例:
注:示例链接
https://gist.github.com/nkreeger/01b5386b522b0cd1f22bc864320f3084#file-baseball-training-data-sample-csv
以下是针对打击区域绘制的训练数据的样子。蓝点标记为坏球,橙点标记为好球(此为大联盟裁判员称谓):
利用 tensorflow.js 构建模型
tensorflow.js 将机器学习引入 javascript 和 web。 我们将利用这个很棒的框架来构建一个深度神经网络模型。这个模型将能够按大联盟裁判的精准度来称呼好球和坏球。
输入 input
该模型在 pitchf / x 的以下字段中进行了训练:
协调球越过本垒的位置('px'和'pz')。
击球手站在垒的哪一侧。
击球区(击球手的躯干)的高度,以英尺为单位。
击球区底部的高度(击球手的膝盖)以英尺为单位。
裁判所称的投球(好球或坏球)的实际标签。
结构 architecture
该模型将通过使用 tensorflow.js 图层 api 定义。layers api 基于 keras,对以前使用过该框架的人来说应该很熟悉:
1const model = tf.sequential();
2
3// two fully connected layers with dropout between each:
4model.add(tf.layers.dense({units: 24, activation: 'relu', inputshape: [5]}));
5model.add(tf.layers.dropout({rate: 0.01}));
6model.add(tf.layers.dense({units: 16, activation: 'relu'}));
7model.add(tf.layers.dropout({rate: 0.01}));
8
9// only two classes: strike and ball:
10model.add(tf.layers.dense({units: 2, activation: 'softmax'}));
11
12model.compile({
13optimizer: tf.train.adam(0.01),
14loss: 'categoricalcrossentropy',
15metrics: ['accuracy']
16});
加载和准备数据
精选的训练集可通过github gist 获得。需要下载此数据集才能开始将 csv 数据转换为 tensorflow.js 用于训练的格式。
注:github gist 链接
https://gist.github.com/nkreeger/43edc6e6daecc2cb02a2dd3293a08f29
1const data = [];
2csvdata.foreach((values) => {
3// 'logit' data uses the 5 fields:
4const x = [];
5x.push(parsefloat(values.px));
6x.push(parsefloat(values.pz));
7x.push(parsefloat(values.sz_top));
8x.push(parsefloat(values.sz_bot));
9x.push(parsefloat(values.left_handed_batter));
10// the label is simply 'is strike' or 'is ball':
11const y = parseint(values.is_strike, 10);
12data.push({x: x, y: y});
13});
14// shuffle the contents to ensure the model does not always train on the same
15// sequence of pitch data:
16tf.util.shuffle(data);
解析 csv 数据后,需要将 js 类型转换为 tensor 批次进行培训和评估。有关此过程的详细信息,请参阅代码实验室。tensorflow.js 团队正在开发一种新的 data api,以便将来更容易获取。
注:代码实验室
https://beta.observablehq.com/@nkreeger/visualizing-ml-training-using-tensorflow-js-and-baseball-d#batches
训练模型
让我们把这一切都整合在一起吧。定义了模型,准备好了训练数据,现在我们已经准备好开始训练了。以下异步方法训练一批训练样本并更新热图:
1// trains and reports loss+accuracy for one batch of training data:
2async function trainbatch(index) {
3const history = await model.fit(batches[index].x, batches[index].y, {
4epochs: 1,
5shuffle: false,
6validationdata: [batches[index].x, batches[index].y],
7batchsize: constants.batch_size
8});
9
10// don't block the ui frame by using tf.nextframe()
11await tf.nextframe();
12updateheatmap();
13await tf.nextframe();
14}
可视化模型的准确性
使用来自均匀放置在本垒板上方的 4 x 4 英尺栅格的预测矩阵来构建热图。在每个训练步骤之后将该矩阵传递到模型中以检查模型的准确度。使用 d3 库将该预测的结果呈现为热图。
构建预测矩阵
热图中使用的预测矩阵从本垒板的中间开始,向左和向右各延伸 2 英尺。它的范围也从本垒板的底部到 4 英尺高。击打区样本位于本垒板上方 1.5 至 3.5 英尺之间。下图有助于让这些 2d 窗格可视化:
该视觉显示了打击区域和预测矩阵与本垒板和游戏区域相关的位置
将预测矩阵与模型一起使用
每个批次在模型中训练之后,预测矩阵被传递到模型中用以请求矩阵中的好球或坏球预测:
1function predictzone() {
2const predictions = model.predictonbatch(predictionmatrix.data);
3const values = predictions.datasync();
4
5// sort each value so the higher prediction is the first element in the array:
6const results = [];
7let index = 0;
8for (let i = 0; i b.value - a.value);
13results.push(list);
14}
15return results;
16}
热图与 d3
现在可以使用 d3 显示预测结果。 来自 50x50 网格中的每一个元素将在 svg 中呈现为 10px x 10px 的矩形。每个矩形的颜色取决于预测结果(好球或者坏球)以及模型对该结果的确定程度(范围从 50%-100%)。 以下代码段显示了如何从 d3 svg 矩形分组更新数据:
1function updateheatmap() {
2rects.data(generateheatmapdata());
3rects
4.attr('x', (coord) => { return scalex(coord.x) * constants.heatmap_size; })
5.attr('y', (coord) => { return scaley(coord.y) * constants.heatmap_size; })
6.attr('width', constants.heatmap_size)
7.attr('height', constants.heatmap_size)
8.style('fill', (coord) => {
9if (coord.strike) {
10return strikecolorscale(coord.value);
11} else {
12return ballcolorscale(coord.value);
13}
14});
15}
有关使用 d3 绘制热图的完整详细信息,请参阅此部分。
注:此部分链接
https://beta.observablehq.com/@nkreeger/visualizing-ml-training-using-tensorflow-js-and-baseball-d#colordomain
总结
网络上有许多令人惊叹的第三方库和工具,可用于创建视觉效果。将这些与机器学习的强大功能与 tensorflow.js 相结合,开发人员能够创建一些非常新奇有趣的演示。
LTC2383-16:具串行接口的16 位、1Msps、低功
最新消息:被我国捕获的美国无人潜航器今天中午顺利移交
视觉焊缝跟踪器的原理和优点
Microchip开发工具屡获殊荣 持续发展壮大
全球服务器出货量将在第三季度反弹
利用TensorFlow.js,D3.js 和 Web 的力量使训练模型的过程可视化
8个LED及照明相关项目入选中国专利优秀奖
电源基础电路图合集免费下载
华为将在2018年前投资6亿美元用于5G研究与创新
舍弗勒展示电气化动力总成解决方案 创新底盘应用及未来移动出行
超精密加工的发展阶段
iPhone7s配置曝光:5英寸屏+纵向双摄 或将采用OLED屏幕
安徽省农业科学院选购我司HS-DSC-101差示扫描量热仪
gsm主要组成是什么_gsm系统结构是怎样的
易充新能源陆续智能充电堆产品极大提高使用效率
实现软件定义工厂:超越工业4.0,从恩智浦N-AFE可配置模拟前端起步!
IPv6协议对安防行业的发展有着怎样的帮助
CAD使用心得之二:CAD图层如何设置?
基于STC12C5A60S2的6路抢答器设计
了解地磁传感器的工作原理