【亲测免费】 Brain.js 神经网络库使用教程

2026-01-15 17:26:14作者：秋阔奎Evelyn

1. 项目介绍

Brain.js 是一个用 JavaScript 编写的简单前馈神经网络库。它旨在帮助开发者在浏览器和 Node.js 环境中快速实现神经网络功能。Brain.js 提供了易于使用的 API，使得训练和运行神经网络变得简单。

2. 项目快速启动

2.1 安装

在 Node.js 环境中，你可以使用 npm 安装 Brain.js：

npm install brain.js

2.2 基本使用

以下是一个简单的示例，展示了如何使用 Brain.js 训练一个神经网络来近似 XOR 函数：

const brain = require('brain.js');

// 创建一个新的神经网络实例
const net = new brain.NeuralNetwork();

// 训练网络
net.train([
  { input: [0, 0], output: [0] },
  { input: [0, 1], output: [1] },
  { input: [1, 0], output: [1] },
  { input: [1, 1], output: [0] }
]);

// 运行网络
const output = net.run([1, 0]);
console.log(output); // 输出: [0.987]

2.3 在浏览器中使用

在浏览器中使用 Brain.js 时，你可以直接引入 brain.js 文件：

<script src="https://cdn.jsdelivr.net/npm/brain.js"></script>
<script>
  const net = new brain.NeuralNetwork();
  net.train([
    { input: [0, 0], output: [0] },
    { input: [0, 1], output: [1] },
    { input: [1, 0], output: [1] },
    { input: [1, 1], output: [0] }
  ]);
  const output = net.run([1, 0]);
  console.log(output); // 输出: [0.987]
</script>

3. 应用案例和最佳实践

3.1 颜色对比识别

Brain.js 可以用于训练神经网络来识别颜色对比。例如，你可以训练一个网络来判断某个颜色是更适合黑色文字还是白色文字：

const net = new brain.NeuralNetwork();

net.train([
  { input: { r: 0.03, g: 0.7, b: 0.5 }, output: { black: 1 } },
  { input: { r: 0.16, g: 0.09, b: 0.2 }, output: { white: 1 } },
  { input: { r: 0.5, g: 0.5, b: 1.0 }, output: { white: 1 } }
]);

const output = net.run({ r: 1, g: 0.4, b: 0 });
console.log(output); // 输出: { white: 0.99, black: 0.002 }

3.2 自定义训练选项

你可以通过设置训练选项来优化网络的训练过程：

net.train(data, {
  errorThresh: 0.005, // 错误阈值
  iterations: 20000,  // 最大训练迭代次数
  log: true,          // 是否记录训练进度
  logPeriod: 10,      // 记录间隔
  learningRate: 0.3   // 学习率
});