TensorFlow.js 是专为前端优化的轻量级 ML 库,支持浏览器实时推理、迁移学习、从零训练及加载 Python 模型;三步可跑通 MNIST 示例,实用中需注意模型大小、转换工具和后端加速。
JavaScript 本身不内置机器学习能力,但通过 TensorFlow.js(简称 TF.js),你可以在浏览器或 Node.js 环境中直接用 JS 训练、加载和运行机器学习模型——无需 Python,也不用部署后端服务。
它不是“JS 版 TensorFlow”,而是专为前端优化的轻量级 ML 库,支持:
以浏览器中识别手写数字(MNIST)为例:
.layers.dense({inputShape: [784], units: 128, activation: 'relu'}));
model.add(tf.layers.dense({units: 10, activation: 'softmax'}));
model.compile({optimizer: 'adam', loss: 'categoricalCrossentropy', metrics: ['accuracy']});
await model.fit(trainImages, trainLabels, {epochs: 5});
真实项目中容易卡住的地方:
基本上就这些。不用重学算法,重点是把数据喂对形状(tensor)、选对层类型、理解训练循环逻辑。TF.js 文档示例丰富,从“Hello World”回归到 GAN 都有现成代码可改。