1

我正在尝试使用 ml5js 制作一些简单的 nn。我有一堆 json 格式的数据,16 个输入(数字从 0 到 15)和 1 个输出(数字从 1 到 4)https://raw.githubusercontent.com/dphdmn/aiTest/main/9473_moves_data.json

然后我正在制作模型并加载数据:

let model;
let options = {
    inputs: 16,
    outputs: 1,
    task: 'classification',
    debug: 'true'
  };
model = ml5.neuralNetwork(options);
//loading my data
model.loadData('9473_moves_data.json', modelLoaded);

然后我正在尝试训练模型,但失败了:

function modelLoaded(){
    console.log(model.data); //it shows the data successfully, so it is loaded!
    console.log('starting training');
    model.normalizeData(); //normalazing data to 0..1
    let options = {
      epochs: 20
    }//start training
    model.train(options, whileTraining, finishedTraining); 
}
//train log
function whileTraining(epoch, loss) {
  console.log(epoch);
}
function finishedTraining() {
  console.log('finished training.');
}

如果我使用任务:'分类',它在训练时什么都不做,输出只是“开始训练”,“完成训练”,就像根本没有训练一样!

如果我将任务更改为“回归”,它开始工作,但不是很有希望,它只是停留在 0.14 损失值,之后,对于任何数组示例,它给出预测“2”,即使在训练数据上也是如此。

4

0 回答 0