javascript - 如何使用 csv 文件训练 tensorflow.js 模型？

Question

是否可以使用我计算机上 csv 文件的路径来训练 tensorflow.js 模型？谢谢!

Answer 1

0.14.1版本开始，可以使用tf.data.csv读取csv文件.但是数据应该由服务器发送。

可以使用像 http-server 这样的服务器来提供本地文件。 .

http-server -c1 --cors .

或者使用python3 http-server

python3 -m http.server .

然后你就可以读取你的csv文件了

const csvUrl = 'localhost:port/file'; 

async function run() {

   const csvDataset = tf.data.csv(
     csvUrl, {
       columnConfigs: {
         medv: {
           isLabel: true
         }
       }
     });

// then you can use your dataset

}