欢迎访问 TensorFlow.js 官方文档!这是一个用于浏览器和Node.js的JavaScript库,可帮助你构建和训练机器学习模型。🧠

什么是 TensorFlow.js?

TensorFlow.js 是谷歌开发的开源库,支持在浏览器和Node.js环境中运行机器学习模型。它结合了机器学习和Web技术,让开发者能够直接在客户端进行模型训练和推理。📦

核心功能

  • 模型训练:在浏览器中训练模型(无需服务器)
  • 神经网络构建:使用Keras API创建模型
  • API接口:支持TensorFlow.js的JavaScript API
  • 扩展阅读TensorFlow.js 官方文档首页 提供完整教程

快速入门示例 💻

以下是一个简单的神经网络示例:

// 加载模型
(async () => {
  const model = tf.sequential();
  model.add(tf.layers.dense({inputShape: [1], units: 1}));
  await model.compile({optimizer: 'sgd', loss: 'meanSquaredError'});
  // 训练模型
  await model.fit([0, 1, 2, 3], [0, 1, 2, 3], {epochs: 10});
})();

👉 点击这里了解完整教程

图片展示

TensorFlow_js