如何解决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 举报,一经查实,本站将立刻删除。