ml5.js模型的培训无法开始

问题描述

我尝试通过LSTM层来预测股票价格。

这是代码：

var options = {
     task: "regression",debug: true,inputs: ["date","timevalue"],outputs: ["price",layers: [
          {
               type: 'lstm',units: 1,inputShape: [10048,2],activation: 'tanh',useBias: true,return_sequences: true,},{
               type: 'dense',inputShape: [1],],};
     

var nn = ml5.neuralNetwork(options);
setData();

async function getData(){
     var data = await fetch("apple_stock.json");
     data = await data.json();
     var cleaned = await data.map( (entry,i) => { 
          var date = entry.Date.split("-");
          date = new Date(date[0],date[1],date[2]).getTime();
          var result = {
               "date": date,"price": entry.High,"tval": i,};
          return result;
     }).filter( result => (result.date != "" || result.date != undefined) && (result.price != "" || result.price != undefined) );
     return cleaned;
}

async function setData() {
     var obj = await getData();
     var tval = 1;
     obj.forEach(item => {
          var input = { "date": parseInt(item.date),"timevalue": ++item.tval };
          var output = { "price": parseInt(item.price),"timevalue": item.tval };
          nn.addData(input,output);
     });
     console.log(obj);
     nn.normalizeData();
     
     train();
}

function train() {
     var trainingOptions = {
          epochs: 256,batchSize: 1024,};
     
     nn.train(trainingOptions,predict);
     console.log(nn.data);
}

function predict(){
     /* nn.predict([ parseInt(new Date(2020,10,17).getTime()) ]).then((result) => {
          console.log(result);
     }); */
     
     //nn.save();
     console.log("end");
}

我希望它可以开始模型训练，但是它什么也没做。我在控制台上没有收到任何错误输出，而且json的数据也已正确加载。有人可以帮忙吗？

我尝试为inputShape使用不同的值，但这没有帮助。我肯定知道所有代码都是从头到尾运行的。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）