如何解决ML5 神经网络的基本训练不起作用
我正在使用 ML5 来训练神经网络。 我正在加载一个包含泰坦尼克号数据的 CSV 文件。 This works when I download the demo file from the ML5 GitHub.
但是当我使用不同的 CSV 文件并替换代码中的列名时,它停止工作。我错过了什么吗?我的 CSV 文件包含数字,而演示文件包含字符串是不是有问题?
let neuralNetwork
function start() {
const nnoptions = {
dataUrl: "data/titanic.csv",inputs: ["Pclass","Sex","Age","SibSp"],// labels from my CSV file
outputs: ["Survived"],task: "classification",debug: true,};
neuralNetwork = ml5.neuralNetwork(nnoptions,modelReady);
}
function modelReady() {
neuralNetwork.normalizeData();
neuralNetwork.train({ epochs: 50 },whileTraining,finishedTraining);
}
// this doesn't get called at all
function whileTraining(epoch,logs) {
console.log(`Epoch: ${epoch} - loss: ${logs.loss.toFixed(2)}`);
}
// this gets called immediately
function finishedTraining() {
console.log("done!");
}
start()
控制台立即显示“完成!”,但模型未经过训练。没有错误信息。 奇怪的是,当标签名称不正确时,我确实收到错误消息。所以标签名称实际上是被识别的。
原始 CSV 文件,有效:
survived,fare_class,sex,age,fare
died,first,male,39,0
died,29,0
我的 CSV 文件,不起作用:
Survived,Pclass,Sex,Age,SibSp,Parch,fare,Embarked
0,3,1,22.0,7.25,1
1,38.0,71.2833,2
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。