我正在尝试使用 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”,即使在训练数据上也是如此。