微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

使用 ml5js

如何解决使用 ml5js

我正在尝试使用 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.');
}

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

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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。