如何使用csv文件训练tensorflow.js模型?

时间:2018-11-29 05:30:19

标签: javascript machine-learning tensorflow.js

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

3 个答案:

答案 0 :(得分:2)

从版本 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

}

答案 1 :(得分:0)

您可以从这里开始:https://js.tensorflow.org/api/0.14.1/#data.csv。 但是不确定是否可以从本地文件存储中提取csv。但是你尝试了什么?

做完一些工作后再回来!

答案 2 :(得分:0)

您还可以使用在线转换器将数据转换为JSON文件。这可能是最简单的方法。