ML5神经网络不工作的基本训练

Question

我正在使用ML5来训练神经网络。我正在加载一个包含泰坦尼克号数据的CSV文件。This works when I download the demo file from the ML5 GitHub.

但是，当我使用不同的CSV文件，并替换了代码中的列名时，它就停止工作了。我是不是遗漏了什么？我的CSV文件包含数字，而demo文件包含字符串，这有问题吗？

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,first,male,29,0

我的CSV文件不工作：

Survived,Pclass,Sex,Age,SibSp,Parch,Fare,Embarked
0,3,1,22.0,1,0,7.25,1
1,1,0,38.0,1,0,71.2833,2

Answer 1

以防有人遇到这个问题:当你分类时，标签总是必须是一个字符串……

工作中的CSV文件：

Survived,Pclass,Sex,Age,SibSp,Parch,Fare,Embarked
yes,3,1,22.0,1,0,7.25,1
no,1,0,38.0,1,0,71.2833,2