Brain.js 可以做回归吗？

Question

我正在尝试使用访问者数量、类别、年龄等属性来定义网站的价格......但事情是线性回归不够准确，所以我想知道是否可以训练大脑 js 做一些某种回归？

我有一个网站列表，其中包含所有属性及其价格来训练 NN，最后只需发送我希望 NN 向我返回价格的属性

Answer 1

我通过 `github` 问题整理了您在 `brain.js` 中的回归可能性。

您所要做的就是在[0-1]范围内表示您的每个input和output

---

第一个组合将是年龄，即从 0 到 1

让0.10表示10岁，记住age cannot exceed more than 1.0

---

第二种组合是从 0 到 1 每小时/每天/每周的访问次数

让0.25表示每小时访问25次，或者您可以根据自己每小时/每天/每周访问的需要等来表示0.25 。

---

第三个组合将属于不同类别，每个具有从 0 到 1 的不同值

让0.01代表第一类，例如sports 。

---

第四种组合是 output 即price每个具有从 0 到 1 的不同值

让0.30代表price等于30美元。

---

代码

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

const network = new brain.NeuralNetwork();

network.train([
        // [age, visits, category]     [price]
        
  { input: [0.10, 0.20, 0.40], output: [0.30] }, // 1 input
  { input: [0.10, 0.50, 0.50], output: [0.60] }, // 2 input
  { input: [0.10, 0.40, 0.60], output: [0.20] }, // 3 input
  { input: [0.10, 0.40, 0.50], output: [0.50] }, // 4 input
  { input: [0.10, 0.30, 0.30], output: [0.90] }, // 5 input
  { input: [0.10, 0.70, 0.30], output: [0.70] }  // 6 input
]);


//multiply answer by 100 to get price value 
const output = network.run([0.10, 0.50, 0.60])*100;

console.log(`Price: ${output}`); // output lies between 2 and 3 input         
                                                          i-e 38.1325