tensorflow.js - Tensorflow.js 数据集到张量？

Question

当底层“数据示例”是平面数组时，是否有推荐/有效的方法将 a 转换tf.data.Dataset为 a ？TensorDataset

我tf.data.csv用于读取和解析 CSV，但随后想使用 Tensorflow.js Core API 将数据处理为tf.Tensors.

标签： tensorflow.js

Answer 1

请注意，通常不建议使用此工作流程，因为将主 JavaScript 内存中的所有数据具体化可能不适用于大型 CSV 数据集。

您可以使用对象的toArray()方法tf.data.Dataset。例如：

  const csvUrl =
'https://storage.googleapis.com/tfjs-examples/multivariate-linear-regression/data/boston-housing-train.csv';

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

  const tensors = await csvDataset.toArray();
  console.log(tensors.length);
  console.log(tensors[0][0]);