【问题标题】:How to train a tensorflow.js model using a csv file?如何使用 csv 文件训练 tensorflow.js 模型?
【发布时间】:2018-11-29 05:30:19
【问题描述】:
是否可以使用我计算机上 csv 文件的路径来训练 tensorflow.js 模型?谢谢!
【问题讨论】:
标签:
javascript
machine-learning
tensorflow.js
【解决方案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
}
【讨论】:
-
我们的tf.data.csv() 便捷方法确实假设了一个URL 源;它只是return new CSVDataset(new URLDataSource(source), csvConfig);。使用稍低级别的 API,您可以在此处使用 FileDataSource,将文件从浏览器上传(参见 stackoverflow.com/questions/53639919/…)传递给它。
【解决方案3】:
您还可以使用在线转换器将数据转换为 JSON 文件。这可能是最简单的方法。