首页 > 解决方案 > 有没有办法让 tensorflow js 从 json 文件而不是 csv 读取/训练数据?

问题描述

所以我是机器学习的新手,并且一直在努力根据 json 文件提供的数据制作我的模型训练。我用 csv 训练没有问题。将 json 转换为 csv 也不是一个选项,因为 json 文件并不简单。

标签: javascriptjsonmachine-learningtraining-datatensorflow.js

解决方案


目前,tensorflow.js 不提供加载 json 数据进行训练的方法。所以有选项

  • 使用 json 到 csv 转换器

这种方式tf.data.csv可以用来读取文件

  • 使用自定义加载器

例如,第一个需要使用 fetch API 获取数据。然后,需要使用返回的数据构建数据集。当数据很大时,不需要同时将所有内容加载到内存中。这个答案讨论了如何处理这种情况。


推荐阅读